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)
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.
|
View source on GitHub