I need to update the net’s parameters according to my own loss function. In that loss function, I have the term containing the norm of a vector, which is the multiplication of the jacobi matrix (output w.r.t paramter) and parameter vector.
How can I get the jacobi and parameters and multiply them together to create loss?