ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

tf.raw_ops.BatchMatrixSetDiag

input A Tensor.
diagonal A Tensor. Must have the same type as input.
name A name for the operation (optional).

A Tensor. Has the same type as input.