is there an efficient way of getting jacobian of every layer in pytorch.
for eg. if i have x=nn.conv2d(y), i would like to see the local gradient dx/dy and similarly so for all the layers in the network.