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