torch.autograd.Function zeroes out (n-1) out of n incoming gradients for single input x and n outputs

I assume the gradients for out1 and out2 are not calculated, as they both reference to out3.
If you add a clone() operation to out2 = out1.clone() and out3, they will be treated differently.
Probably related to this post.

1 Like