Gradient of the convex potential w.r.t. input.
For scalar output (output_dim == 1), returns the gradient.
For vector output, returns the Jacobian.
- Parameters:
x (Array) – Input of shape [batch, input_dim].
- Return type:
Array
- Returns:
Gradients of shape [batch, input_dim] (scalar output) or
[batch, output_dim, input_dim] (vector output).
