description: TensorFlow variant of NumPy's diag_indices.

tf.experimental.numpy.diag_indices

View source on GitHub

TensorFlow variant of NumPy's diag_indices.

tf.experimental.numpy.diag_indices(
    n, ndim=2
)

See the NumPy documentation for numpy.diag_indices.