sponge.function.pbc_image

查看源文件
sponge.function.pbc_image(position: Tensor, pbc_box: Tensor, offset: float = 0)[源代码]

计算PBC box的周期图。

参数:
  • position (Tensor) - 位置坐标,数据类型为float,shape为 \((B, ..., D)\) ,B是Batch size,D是模拟系统的维度,一般为3。

  • pbc_box (Tensor) - PBC box,数据类型为float,shape为 \((B, D)\)

  • offset (float) - 偏移率。默认值: 0.0

返回:

Tensor。周期图,shape为 \((B, ..., D)\) ,数据类型为int32。

支持平台:

Ascend GPU