W3cubDocs

/TensorFlow Python

tf.contrib.framework.smart_cond

tf.contrib.framework.smart_cond(
    pred,
    true_fn=None,
    false_fn=None,
    name=None
)

Defined in tensorflow/python/framework/smart_cond.py.

Return either true_fn() if predicate pred is true else false_fn().

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.

© 2018 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/api_docs/python/tf/contrib/framework/smart_cond