tf.raw_ops.StatelessIf | TensorFlow v2.16.1 (original) (raw)

tf.raw_ops.StatelessIf

Stay organized with collections Save and categorize content based on your preferences.

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

Compat aliases for migration

tf.compat.v1.raw_ops.StatelessIf

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

Args
cond	A Tensor. A Tensor. If the tensor is a scalar of non-boolean type, the scalar is converted to a boolean according to the following rule: if the scalar is a numerical value, non-zero meansTrue and zero means False; if the scalar is a string, non-empty means True and empty means False. If the tensor is not a scalar, being empty means False and being non-empty means True.This should only be used when the if then/else body functions do not have stateful ops.
input	A list of Tensor objects. A list of input tensors.
Tout	A list of tf.DTypes. A list of output types.
then_branch	A function decorated with @Defun. A function that takes 'inputs' and returns a list of tensors, whose types are the same as what else_branch returns.
else_branch	A function decorated with @Defun. A function that takes 'inputs' and returns a list of tensors, whose types are the same as what then_branch returns.
output_shapes	An optional list of shapes (each a tf.TensorShape or list of ints). Defaults to [].
name	A name for the operation (optional).

Returns
A list of Tensor objects of type Tout.

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license.

Last updated 2024-04-26 UTC.