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 | Условие. A 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. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/python/tf/raw_ops/StatelessIf