tf.compat.v1.assert_rank_in
Проверка, что x имеет ранг в ranks.
tf.compat.v1.assert_rank_in(
x, ranks, data=None, summarize=None, message=None, name=None
)
Пример добавления зависимости к операции:
with tf.control_dependencies([tf.compat.v1.assert_rank_in(x, (2, 4))]): output = tf.reduce_sum(x)
| Аргументы | |
|---|---|
x | Численное значение Tensor. |
ranks | Итерируемый набор скалярных Tensor объектов. |
data | Тензоры для вывода, если условие ложно. По умолчанию отображается сообщение об ошибке и первые несколько элементов x. |
summarize | Количество элементов каждого тензора для вывода. |
message | Строка для добавления к стандартному сообщению. |
name | Имя этой операции (необязательно). По умолчанию "assert_rank_in". |
| Возвращаемое значение | |
|---|---|
Операция, генерирующая InvalidArgumentError если ранг x не принадлежит к ranks. Если статические проверки определяют, что x имеет соответствующий ранг, возвращается no_op. |
| Исключения | |
|---|---|
ValueError | Если статические проверки определяют, что x имеет несоответствующий ранг. |
© 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/versions/r2.9/api_docs/python/tf/compat/v1/assert_rank_in