tf.contrib.framework.smart_case
Like tf.case, except attempts to statically evaluate predicates.
tf.contrib.framework.smart_case(
pred_fn_pairs, default=None, exclusive=False, name='smart_case'
)
If any predicate in pred_fn_pairs is a bool or has a constant value, the associated callable will be called or omitted depending on its value. Otherwise this functions like tf.case.
| Args | |
|---|---|
pred_fn_pairs | Dict or list of pairs of a boolean scalar tensor and a callable which returns a list of tensors. |
default | Optional callable that returns a list of tensors. |
exclusive | True iff at most one predicate is allowed to evaluate to True. |
name | A name for this operation (optional). |
| Returns | |
|---|---|
The tensors returned by the first pair whose predicate evaluated to True, or those returned by default if none does. |
| Raises | |
|---|---|
TypeError | If pred_fn_pairs is not a list/dictionary. |
TypeError | If pred_fn_pairs is a list but does not contain 2-tuples. |
TypeError | If fns[i] is not callable for any i, or default 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_case