tf.raw_ops.If

output = cond ? then_branch(input) : else_branch(input)

Просмотреть псевдонимы

Псевдонимы совместимости для миграции

Подробнее см. в руководстве по миграции.

tf.compat.v1.raw_ops.If

tf.raw_ops.If(
    cond, input, Tout, then_branch, else_branch, output_shapes=[], name=None
)

Аргументы
`cond`	`Tensor`. Тензор. Если тензор является скаляром не логического типа, скаляр преобразуется в логическое значение по следующему правилу: если скаляр является числовым значением, ненулевое значение означает `True`, а ноль означает False; если скаляр является строкой, непустая строка означает `True`, а пустая означает `False`. Если тензор не является скаляром, пустое значение означает False, а непустое — True.
`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/If