jax-js
    Preparing search index...

    Function vmap

    • Vectorize an operation on a batched axis for one or more inputs.

      Type Parameters

      Parameters

      • f: F
      • OptionalinAxes: number | MappedJsTree<Parameters<F>, ArrayLike, number | null>

      Returns (...args: MappedJsTree<Parameters<F>, Array, ArrayLike>) => ReturnType<F>