tf.compat.v1.assert_rank_in
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. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/python/tf/compat/v1/assert_rank_in