tf.experimental.dtensor.check_layout

Asserts that the layout of the DTensor is layout.

tf.experimental.dtensor.check_layout(
    tensor: tf.Tensor,
    layout: tf.experimental.dtensor.Layout
) -> None

Args
tensor	A DTensor whose layout is to be checked.
layout	The Layout to compare against.

Raises
ValueError	If the layout of tensor does not match the supplied layout.