description: TensorFlow variant of NumPy's diagonal.

tf.experimental.numpy.diagonal

View source on GitHub

TensorFlow variant of NumPy's diagonal.

tf.experimental.numpy.diagonal(
    a, offset=0, axis1=0, axis2=1
)

See the NumPy documentation for numpy.diagonal.