tf.raw_ops.BatchMatrixDiagPart

inputA Tensor.
nameA name for the operation (optional).

A Tensor. Has the same type as input.