tf.experimental.numpy.diagonal

TensorFlow variant of NumPy's diagonal.

See the NumPy documentation for numpy.diagonal.