Closed slerman12 closed 3 years ago
My model outputs a vector, but I actually only need the grad for one scalar in that vector. Is there a way I can get this to work?
My model outputs a vector, but I actually only need the grad for one scalar in that vector. Is there a way I can get this to work?