Solved by Diego in post #3 
              
              
                Hi, so if by sensitivity you mean, how sensible is the output to changes in the input (e.g the gradients). Then something like this could work: 
>>> inp = torch.randn(5)
>>> 

>>> inp.requires_grad_()
tensor([-0.1457,  0.1341,  0.6884, -1.9446, -0.2557], requires_grad=True)
>>> x = inp * 7 
>>> x = …