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. |
© 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/keras/backend/switch