jax-js
System
Light
Dark
Search…
Preparing search index...
@jax-js/jax
tree
map
Function map
map
<
T
,
U
,
Tree
>
(
fn
:
(
...
args
:
T
[]
)
=>
U
,
tree
:
Tree
,
...
rest
:
Tree
[]
,
)
:
MapJsTree
<
Tree
,
T
,
U
>
Maps a multi-input function over pytree args to produce a new pytree.
Type Parameters
T
U
Tree
Parameters
fn
:
(
...
args
:
T
[]
)
=>
U
tree
:
Tree
...
rest
:
Tree
[]
Returns
MapJsTree
<
Tree
,
T
,
U
>
jax-js
Loading...
Maps a multi-input function over pytree args to produce a new pytree.