コード例 #1
0
ファイル: space.py プロジェクト: zizai/jax-md
 def shift(R, dR, t=None, **unused_kwargs):
   _check_time_dependence(t)
   check_kwargs_empty(unused_kwargs)
   return periodic_shift(f32(1.0), R, transform(_small_inverse(T(t)), dR))
コード例 #2
0
ファイル: space.py プロジェクト: zizai/jax-md
 def shift(R, dR, t=None, **unused_kwargs):
   _check_time_dependence(t)
   check_kwargs_empty(unused_kwargs)
   return R + transform(_small_inverse(T(t)), dR)
コード例 #3
0
ファイル: space.py プロジェクト: zizai/jax-md
 def displacement(Ra, Rb, t=None, **unused_kwargs):
   _check_time_dependence(t)
   check_kwargs_empty(unused_kwargs)
   dR = periodic_displacement(f32(1.0), pairwise_displacement(Ra, Rb))
   return transform(T(t), dR)