tvdw:

    gradient = gradient[0].tolist()
    gradient = gradient[0]



What if we want to store the gradient of output w.r.t input for every epoch? Does this code work too? why in this code there is no loss.backward()?