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