Function vmap

vmap<F extends (...args: any[]) => JsTree<Array>>(
f: F,
inAxes?: number | JsTree<number | null>[],
): (...args: MappedJsTree<Parameters<F>, Array, ArrayLike>) => ReturnType<F>
Vectorize an operation on a batched axis for one or more inputs.
Type Parameters
- F extends (...args: any[]) => JsTree<Array>
Parameters
- f: F
- OptionalinAxes: number | JsTree<number | null>[]
Returns (...args: MappedJsTree<Parameters<F>, Array, ArrayLike>) => ReturnType<F>
- Defined in src/index.ts:63