tf.raw_ops.BatchMatrixDiag

diagonal A Tensor.
name A name for the operation (optional).

A Tensor. Has the same type as diagonal.