View source on GitHub |
Assert x
has rank in ranks
.
tf.compat.v1.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)
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".Op raising InvalidArgumentError
unless rank of x
is in ranks
.
If static checks determine x
has matching rank, a no_op
is returned.
ValueError
: If static checks determine x
has mismatched rank.