Hi Kadek!



 gemsanyou:

However, I also need the gradient to be a vector, with each column is the gradient of each column of the Loss with respect to the parameters


Does autograd’s jacobian() 3 do what you want?
Best.
K. Frank