I found the gradient of weight using this method and it matched the pytorch one but the gradient of the input doesn’t match the pytorch one. Could you please help me to implement it using fold and unfold.