tfp.edward2.interceptable(func)
Decorator that wraps func so that its execution is intercepted.
The wrapper passes func to the interceptor for the current thread.
If there is no next interceptor, we perform an "immediate" call to func.
That is, func terminates without forwarding its execution to another
interceptor.
Args:
func: Function to wrap.
Returns:
The decorated function.