tf.keras.backend.switch

TensorFlow 2 version View source on GitHub

Switches between two operations depending on a scalar value.

Compat aliases for migration

See Migration guide for more details.

tf.compat.v1.keras.backend.switch, `tf.compat.v2.keras.backend.switch`

 
tf.keras.backend.switch(
    condition, then_expression, else_expression
)

Note that both then_expression and else_expression should be symbolic tensors of the same shape.

condition tensor (int or bool).
then_expression either a tensor, or a callable that returns a tensor.
else_expression either a tensor, or a callable that returns a tensor.

The selected tensor.

ValueError If rank of condition is greater than rank of expressions.