tf.keras.backend.switch
Switches between two operations depending on a scalar value.
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.
| Arguments |
|---|
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. |
| Returns |
|---|
| The selected tensor. |
| Raises |
|---|
ValueError | If rank of condition is greater than rank of expressions. |