jax.numpy.identity
jax.numpy.identity#
- jax.numpy.identity(n, dtype=None)[source]#
Return the identity array.
LAX-backend implementation of
numpy.identity()
.Original docstring below.
The identity array is a square array with ones on the main diagonal.