Eye
===
Returns a 2-D tensor with ones on the diagonal.
**Abstract Signature:**
``Eye(n: int, m: int, dtype: Dtype)``
.. raw:: html
PyTorch
API: torch.eye
Strategy: Direct Mapping
JAX (Core)
API: jax.numpy.eye
Strategy: Direct Mapping
NumPy
API: numpy.eye
Strategy: Direct Mapping
Keras
API: keras.ops.eye
Strategy: Direct Mapping
TensorFlow
API: tf.eye
Strategy: Direct Mapping
Apple MLX
API: mlx.core.eye
Strategy: Direct Mapping
Flax NNX
API: jax.numpy.eye
Strategy: Direct Mapping
PaxML / Praxis
API: jax.numpy.eye
Strategy: Direct Mapping