I used forward_hook this function, it seems only print input and output, if I want to  call the input and output data with variables that should I do? I tried to return input    in the printnorm function, but failed.

> def printnorm(self,input,output):
>     print('Inside '+ self.__class__.__name__ +'forward')
>     print('')
>     print('input: ',input)
> model.conv2.register_forward_hook(printnorm)
> out = model(input)