you need to use hooks if you want to inspect gradients of intermediate variables.
Refer to the discussion here: Why cant I see .grad of an intermediate variable?
In your case you need to attach a hook to out2 and out3, which returns the ‘grad’
you need to use hooks if you want to inspect gradients of intermediate variables.
Refer to the discussion here: Why cant I see .grad of an intermediate variable?
In your case you need to attach a hook to out2 and out3, which returns the ‘grad’