jax-js
System
Light
Dark
Search…
Preparing search index...
@jax-js/jax
vjp
Function vjp
vjp
<
F
extends
(
...
args
:
any
[]
)
=>
JsTree
<
Array
>
>
(
f
:
F
,
...
primals
:
MappedJsTree
<
Parameters
<
F
>
,
Array
,
ArrayLike
>
,
)
:
[
ReturnType
<
F
>
,
(
cotangents
:
MappedJsTree
<
ReturnType
<
F
>
,
Array
,
ArrayLike
>
,
)
=>
MappedJsTree
<
Parameters
<
F
>
,
ArrayLike
,
Array
>
,
]
Calculate the reverse-mode vector-Jacobian product for a function.
Type Parameters
F
extends
(
...
args
:
any
[]
)
=>
JsTree
<
Array
>
Parameters
f
:
F
...
primals
:
MappedJsTree
<
Parameters
<
F
>
,
Array
,
ArrayLike
>
Returns
[
ReturnType
<
F
>
,
(
cotangents
:
MappedJsTree
<
ReturnType
<
F
>
,
Array
,
ArrayLike
>
,
)
=>
MappedJsTree
<
Parameters
<
F
>
,
ArrayLike
,
Array
>
,
]
jax-js
Loading...
Calculate the reverse-mode vector-Jacobian product for a function.