Solved by albanD in post #3 
              
              
                Hi, 

grad_in[2] is the gradients for the bias as you saw.
The conv takes 3 inputs: (input, weights, bias). Since for the first layer, you don’t need gradients for the input, None is returned. For the second layer, gradients for the input are needed to compute the gradients of the first layer and so…