Aliases:
tf.assert_rank_in
tf.debugging.assert_rank_in
tf.debugging.assert_rank_in(
x,
ranks,
data=None,
summarize=None,
message=None,
name=None
)
Defined in tensorflow/python/ops/check_ops.py
.
Assert x
has rank in ranks
.
Example of adding a dependency to an operation:
with tf.control_dependencies([tf.assert_rank_in(x, (2, 4))]):
output = tf.reduce_sum(x)
Args:
x
: NumericTensor
.ranks
: Iterable of scalarTensor
objects.data
: The tensors to print out if the condition is False. Defaults to error message and first few entries ofx
.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 determinex
has mismatched rank.