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`	A `Tensor`. A 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`.

© 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/If