tf.experimental.numpy.trace

View source on GitHub

TensorFlow variant of NumPy's trace.

Unsupported arguments: out.

See the NumPy documentation for numpy.trace.