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