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
submitted by /u/soravoid
[visit reddit] [comments]