Logging gradient w.r.t. input

autograd
1

Hi,
I’m trying to log(tensorboardx) the lossgradient w.r.t. input.
I registered a backward_hook but can’t add a writer .

does somone have tip?