tfp.experimental.distributions.marginal_fns.ps.cond

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.

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.

Tensors returned by the call to either true_fn or false_fn.

TypeError If true_fn or false_fn is not callable.