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.4/api_docs/python/tf/raw_ops/StatelessIf