tf.keras.backend.switch

View source on GitHub

Switches between two operations depending on a scalar value.

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/r1.15/api_docs/python/tf/keras/backend/switch