View source on GitHub
Return either true_fn() if predicate pred is true else false_fn().
tfp.experimental.distributions.marginal_fns.ps.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.
Args | |
|---|---|
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.
|