Join TensorFlow at Google I/O, May 11-12 Register now

tf.raw_ops.BatchMatrixDiagPart

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

A Tensor. Has the same type as input.