Hi All,
Is there a simple way to access and save the grad_input and grad_output as defined in the backward component of a custom torch.autograd.Function? I would like to do this for all the layers and I was wondering if there was something similar to
net.conv11.weight.grad
net.conv21.bias.grad


for the grad_input and grad_output.
Thanks!