TensorFlow 2.0 Beta is available Learn more

tf.debugging.assert_rank_in

TensorFlow 2.0 version View source on GitHub

Assert x has rank in ranks.

Aliases:

  • tf.assert_rank_in
  • tf.compat.v1.assert_rank_in
  • tf.compat.v1.debugging.assert_rank_in
tf.debugging.assert_rank_in(
    x,
    ranks,
    data=None,
    summarize=None,
    message=None,
    name=None
)

Example of adding a dependency to an operation:

with tf.control_dependencies([tf.compat.v1.assert_rank_in(x, (2, 4))]):
  output = tf.reduce_sum(x)

Args:

  • x: Numeric Tensor.
  • ranks: Iterable of scalar Tensor objects.
  • data: The tensors to print out if the condition is False. Defaults to error message and first few entries 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_in".

Returns:

Op raising InvalidArgumentError unless rank of x is in ranks. If static checks determine x has matching rank, a no_op is returned.

Raises:

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