So interestingly enough, the gradient returned in every time step by creating this hook is the same. I’m guessing it’s the sum total of all gradients of z over every time step. Can someone confirm this?