I’m a little new to PyTorch so I realize I am not quite using these methods correctly. I want y.grad to be the gradient dw/dy and the .grad property for all other variables, x_i, to be dz/dx_i. Does that make sense?