tf.raw_ops.BatchMatrixDiagPart

tf.raw_ops.BatchMatrixDiagPart(
    input, name=None
)

Args:

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

Returns:

A Tensor. Has the same type as input.