Use DiCE objective Rewriting the monte-carlo expectation to use the DiCE objective rather than the surrogate loss approach. This ensures that any order gradients of the objective produce unbiased estimates of higher order gradients (including zero order) when not using reparameterization.
Use DiCE objective
Rewriting the monte-carlo expectation to use the DiCE objective rather than the surrogate loss approach. This ensures that any order gradients of the objective produce unbiased estimates of higher order gradients (including zero order) when not using reparameterization.

PiperOrigin-RevId: 199806260

tensorflow/probability,tensorflow/probability