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