tf.linalg.tensor_diag

Returns a diagonal tensor with a given diagonal values.