tf.experimental.numpy.identity

TensorFlow variant of NumPy's identity.

See the NumPy documentation for numpy.identity.