Function jvp

jvp<F extends (...args: any[]) => JsTree<Array>, HA extends boolean = false>(
    f: F,
    primals: MappedJsTree<Parameters<F>, Array, ArrayLike>,
    tangents: MappedJsTree<Parameters<F>, Array, ArrayLike>,
    opts?: { hasAux?: HA },
): HA extends true
    ? ReturnType<F> extends [Out, Aux] ? [Out, Out, Aux] : never
    : [ReturnType<F>, ReturnType<F>]
Compute the forward-mode Jacobian-vector product for a function.
Type Parameters
- F extends (...args: any[]) => JsTree<Array>
- HA extends boolean = false
Parameters
- f: F
- primals: MappedJsTree<Parameters<F>, Array, ArrayLike>
- tangents: MappedJsTree<Parameters<F>, Array, ArrayLike>
- Optionalopts: { hasAux?: HA }
Returns HA extends true
? ReturnType<F> extends [Out, Aux] ? [Out, Out, Aux] : never
: [ReturnType<F>, ReturnType<F>]
- Defined in src/index.ts:45