If I have a tensor x.
Can I define a vector function, say f(x)=(3x, x+2), in such a way that I can then obtain df/dx?
I hope my question is understood. In a nutshell: I want a way to define such a vector function, from which I can get its gradient.
If I have a tensor x.
Can I define a vector function, say f(x)=(3x, x+2), in such a way that I can then obtain df/dx?
I hope my question is understood. In a nutshell: I want a way to define such a vector function, from which I can get its gradient.
Hi Juan!
Does jacobian() do what you want?
If you implement your function using (differentiable) pytorch tensor
functions, autograd and jacobian()
should compute for you the
full set of partial derivatives.
Best.
K. Frank