Function grad

grad<F extends (...args: any[]) => JsTree<Array>>(
f: F,
): (
...primals: MappedJsTree<Parameters<F>, Array, ArrayLike>,
) => MappedJsTree<Parameters<F>[0], ArrayLike, Array>
Compute the gradient of a scalar-valued function f with respect to its first argument.
Type Parameters
- F extends (...args: any[]) => JsTree<Array>
Parameters
- f: F
Returns (
...primals: MappedJsTree<Parameters<F>, Array, ArrayLike>,
) => MappedJsTree<Parameters<F>[0], ArrayLike, Array>
- Defined in index.ts:144