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 | Логическое значение. Это тензор. Если тензор — скаляр небулевого типа, скаляр преобразуется в булевое значение по следующему правилу: если скаляр — числовое значение, ненулевое значение означает 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/versions/r2.9/api_docs/python/tf/raw_ops/StatelessIf