tf.experimental.numpy.diag_indices

TensorFlow variant of NumPy's diag_indices.

See the NumPy documentation for numpy.diag_indices.