|---| | [ View source on GitHub](https://github.com/tensorflow/tensorflow/blob/v2.15.1/tensorflow/python/ops/numpy_ops/np_array_ops.py#L1338-L1350) | TensorFlow variant of NumPy's `diag_indices`. ```python tf.experimental.numpy.diag_indices( n, ndim=2 ) ``` See the NumPy documentation for [`numpy.diag_indices`](https://numpy.org/doc/stable/reference/generated/numpy.diag_indices.html).