So in my classification neural network, my final tensor is
a = tensor([[ 546.3831, -796.4016]], grad_fn=AddmmBackward>). (There is a < in front of Addmm)
I