Save the date! Google I/O returns May 18-20 Register now

tf.experimental.numpy.diag

TensorFlow variant of NumPy's diag.

Raises an error if input is not 1- or 2-d.

See the NumPy documentation for numpy.diag.