jax-js
System
Light
Dark
Search…
Preparing search index...
@jax-js/jax
numpy
identity
Function identity
identity
(
n
:
number
,
__namedParameters
?:
{
device
?:
Device
;
dtype
?:
DType
}
,
)
:
Array
Return the identity matrix, with ones on the main diagonal.
Parameters
n
:
number
__namedParameters
:
{
device
?:
Device
;
dtype
?:
DType
}
= {}
Returns
Array
jax-js
Loading...
Return the identity matrix, with ones on the main diagonal.