jax-js
System
Light
Dark
Search…
Preparing search index...
@jax-js/jax
numpy
trace
Function trace
trace
(
a
:
ArrayLike
,
offset
?:
number
,
axis1
?:
number
,
axis2
?:
number
)
:
Array
Calculate the sum of the diagonal of an array along the given axes.
Parameters
a
:
ArrayLike
offset
:
number
= 0
axis1
:
number
= 0
axis2
:
number
= 1
Returns
Array
jax-js
Loading...
Calculate the sum of the diagonal of an array along the given axes.