コード例 #1
0
ファイル: hmc_test.py プロジェクト: Bonnevie/probability 
 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)
コード例 #2
0
ファイル: hmc_test.py プロジェクト: asudomoeva/probability 
 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)