Hi,
Is there have any method to save a tensor for statistics?
Example
class Exp(Function):
    @staticmethod
     def forward(ctx, i):
         i_max = i.max()
         return i

    @staticmethod
    def backward(ctx, grad_output):
        return grad_output

Thanks!