sponge.function.pbc_image
- sponge.function.pbc_image(position: Tensor, pbc_box: Tensor, offset: float = 0)[source]
calculate the periodic image of the PBC box
- Parameters
position (Tensor) – Tensor of shape
. Data type is float. B is batchsize, i.e. number of walkers in simulation D is spatial dimension of the simulation system. Usually is 3.pbc_box (Tensor) – Tensor of shape
. Data type is float.offset (float) – Offset ratio
relative to box size . Default:0
- Returns
image (Tensor), a tensor of shape
. Data type is int32.
- Supported Platforms:
Ascend
GPU