tf.debugging.assert_rank_in | TensorFlow v2.16.1 (original) (raw)
tf.debugging.assert_rank_in
Stay organized with collections Save and categorize content based on your preferences.
Assert that x has a rank in ranks.
tf.debugging.assert_rank_in(
x, ranks, message=None, name=None
)
This Op checks that the rank of x is in ranks.
If x has a different rank, message, as well as the shape of x are printed, and InvalidArgumentError is raised.
| Args | |
|---|---|
| x | Tensor. |
| ranks | Iterable of scalar Tensor objects. |
| 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. This can be used with tf.control_dependencies inside of tf.functions to block followup computation until the check has executed. |
| Raises | |
|---|---|
| InvalidArgumentError | x does not have rank in ranks, but the rank cannot be statically determined. |
| ValueError | If static checks determine x has mismatched rank. |
eager compatibility
returns None
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.