There is an implementation of conv2d forward pass in the unfold documentation page and it works very wellhelp. Could the same be done for a conv2d backward pass using fold and unfold. If so please help me to implement it. I found out the gradients of weight and bias and it matches the pytorch ones but the gradient of the input seems to be not the same. Please help.
Any solutions for that?