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