description: TensorFlow variant of NumPy's identity.

tf.experimental.numpy.identity

View source on GitHub

TensorFlow variant of NumPy's identity.

tf.experimental.numpy.identity(
    n, dtype=float
)

See the NumPy documentation for numpy.identity.