Hello!
In the work that I’m doing, after the first conv2d() layer, the output is converted to numpy array to do some processing using .detach(). During this process, the new output will be 3 times bigger and then it is converted back to the tensor to be used as a input for the next conv2d() layer.
Is there anyway of getting the gradient back to the new tensor?
Note: The new tensor’s values are just copied from the old tensor.
Thank you guys very much!!!