tf.keras.ops.cond | TensorFlow v2.16.1 (original) (raw)
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. |
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license.
Last updated 2024-06-07 UTC.