tf.assert_rank

Assert that x has rank equal to rank.

tf.assert_rank(
    x,
    rank,
    message=None,
    name=None
)

This Op checks that the rank of x is equal to rank.

If x has a different rank, message, as well as the shape of x are printed, and InvalidArgumentError is raised.

Args:

  • x: Tensor.
  • rank: Scalar integer Tensor.
  • message: A string to prefix to the default message.
  • name: A name for this operation (optional). Defaults to "assert_rank".

Raises:

  • InvalidArgumentError: if the check can be performed immediately and x does not have rank rank. The check can be performed immediately during eager execution or if the shape of x is statically known.

results matching ""

    No results matching ""