tf.raw_ops.StatelessIf
output = cond ? then_branch(input) : else_branch(input)
tf.raw_ops.StatelessIf(
cond, input, Tout, then_branch, else_branch, output_shapes=[], name=None
)
| Аргументы | |
|---|---|
cond | Tensor. Тензор. Если тензор является скаляром не логического типа, скаляр преобразуется в логическое значение по следующему правилу: если скаляр имеет числовое значение, ненулевое значение означает True, а нулевое — False; если скаляр является строкой, непустая строка означает True, а пустая — False. Если тензор не является скаляром, пустой тензор означает False, а непустой — True. Эту операцию следует использовать только в том случае, если функции тела if then/else не содержат операций с состоянием. |
input | Список объектов Tensor. Список входных тензоров. |
Tout | Список tf.DTypes. Список типов выходных данных. |
then_branch | Функция, декорированная с помощью @Defun. Функция, которая принимает «inputs» и возвращает список тензоров, типы которых совпадают с типами, возвращаемыми else_branch. |
else_branch | Функция, декорированная с помощью @Defun. Функция, которая принимает «inputs» и возвращает список тензоров, типы которых совпадают с типами, возвращаемыми then_branch. |
output_shapes | Необязательный список форм (каждая из которых является tf.TensorShape или списком ints). По умолчанию — []. |
name | Имя операции (необязательно). |
| Возвращает | |
|---|---|
Список объектов Tensor типа Tout. |
© 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/raw_ops/StatelessIf