If I am making a function that has two inputs, what values is `tape.gradients` expecting?
For example, if I input data in a batch size to the function `f(x,y)`, the `tape.gradient` expects both of the returned gradients to be of shape (batch size, 2), however I’m not sure what should be the contents of those Tensors. For a batch size of 3, I thought it would look something like the following:
|df/dx (x_1, y_1)||df/dy (x_1, y_1)|
|df/dx (x_2, y_2)||df/dy (x_2, y_2)|
|df/dx (x_3, y_3)||df/dy (x_3, y_3)|
Where the above table is a Tensor, but i have no idea what other tensor it could want for the second variable. Does anyone have an idea of what it should be? Thanks