Ejemplo n.º 1
0
 def leapfrog_one_step(*args):
     return _leapfrog_integrator_one_step(
         lambda x: self._log_gamma_log_prob(x, event_dims),
         independent_chain_ndims, [step_size], *args)
Ejemplo n.º 2
0
 def leapfrog_one_step(*args):
   return _leapfrog_integrator_one_step(
       lambda x: self._log_gamma_log_prob(x, event_dims),
       independent_chain_ndims,
       [step_size],
       *args)