tf.contrib.framework.smart_cond
Return either true_fn() if predicate pred is true else false_fn().
tf.contrib.framework.smart_cond(
pred, true_fn=None, false_fn=None, name=None
)
If pred is a bool or has a constant value, we return either true_fn() or false_fn(), otherwise we use tf.cond to dynamically route to both.
| Arguments | |
|---|---|
pred | A scalar determining whether to return the result of true_fn or false_fn. |
true_fn | The callable to be performed if pred is true. |
false_fn | The callable to be performed if pred is false. |
name | Optional name prefix when using tf.cond. |
| Returns | |
|---|---|
Tensors returned by the call to either true_fn or false_fn. |
| Raises | |
|---|---|
TypeError | If true_fn or false_fn is not callable. |
© 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/contrib/framework/smart_cond