tf.debugging.assert_rank

Aliases:

• tf.assert_rank
• tf.debugging.assert_rank

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

Defined in tensorflow/python/ops/check_ops.py.

Assert x has rank equal to rank.

Example of adding a dependency to an operation:

with tf.control_dependencies([tf.assert_rank(x, 2)]):
  output = tf.reduce_sum(x)

Args:

• x: Numeric Tensor.
• rank: Scalar integer Tensor.
• data: The tensors to print out if the condition is False. Defaults to error message and the shape of x.
• summarize: Print this many entries of each tensor.
• message: A string to prefix to the default message.
• name: A name for this operation (optional). Defaults to "assert_rank".

Returns:

Op raising InvalidArgumentError unless x has specified rank. If static checks determine x has correct rank, a no_op is returned.

Raises:

• ValueError: If static checks determine x has wrong rank.