tf.keras.ops.cond
Conditionally applies true_fn or false_fn.
tf.keras.ops.cond(
pred, true_fn, false_fn
)
| Args | |
|---|---|
pred | Boolean scalar type |
true_fn | Callable returning the output for the pred == True case. |
false_fn | Callable returning the output for the pred == False case. |
| Returns | |
|---|---|
The output of either true_fn or false_fn depending on pred. |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/keras/ops/cond
