tf.debugging.assert_rank
Утверждает, что ранг x равен rank.
tf.debugging.assert_rank(
x, rank, message=None, name=None
)
Эта операция проверяет, что ранг x равен rank.
Если ранг x отличается, выводятся message, а также форма x, и вызывается InvalidArgumentError.
| Аргументы | |
|---|---|
x | Tensor. |
rank | Целое скалярное значение Tensor. |
message | Строка, добавляемая в начало сообщения по умолчанию. |
name | Имя этой операции (необязательно). По умолчанию — "assert_rank". |
| Возвращает | |
|---|---|
Операция, вызывающая InvalidArgumentError, если x не имеет указанного ранга. Если статические проверки определяют, что x имеет правильный ранг, возвращается no_op. Это можно использовать с tf.control_dependencies внутри tf.function, чтобы блокировать последующие вычисления до выполнения проверки. |
| Вызывает исключения | |
|---|---|
InvalidArgumentError | если проверку можно выполнить сразу и x не имеет ранг rank. Проверку можно выполнить сразу при eager-выполнении или если форма x известна статически. |
Совместимость с eager-режимом
возвращает None
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/debugging/assert_rank