jax-js
System
Light
Dark
Search…
Preparing search index...
@jax-js/jax
vmap
Function vmap
vmap
<
F
extends
(
...
args
:
any
[]
)
=>
JsTree
<
Array
>
>
(
f
:
F
,
inAxes
?:
number
|
MappedJsTree
<
Parameters
<
F
>
,
ArrayLike
,
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
Optional
inAxes
:
number
|
MappedJsTree
<
Parameters
<
F
>
,
ArrayLike
,
number
|
null
>
Returns
(
...
args
:
MappedJsTree
<
Parameters
<
F
>
,
Array
,
ArrayLike
>
)
=>
ReturnType
<
F
>
jax-js
Loading...
Vectorize an operation on a batched axis for one or more inputs.