tf.compat.v1.assert_rank_in | TensorFlow v2.16.1 (original) (raw)
tf.compat.v1.assert_rank_in
Stay organized with collections Save and categorize content based on your preferences.
Assert x has rank in ranks.
View aliases
Compat aliases for migration
SeeMigration guide for more details.
tf.compat.v1.debugging.assert_rank_in
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. |
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license.
Last updated 2024-04-26 UTC.