sponge.function.calc_vector

sponge.function.calc_vector(initial: Tensor, terminal: Tensor, pbc_box: Tensor = None)

计算从起点到终点的向量。

参数：

• initial (Tensor) - 起点坐标，shape为 $(...,D)$ 。其中， $D$ 表示模拟系统的维度（通常为3）。

• terminal (Tensor) - 终点坐标，shape为 $(...,D)$ 。

• pbc_box (Tensor) - PBC box，shape为 $(D)$ 或 $(B,D)$ 。其中，$B$ 为batch size。默认为 None 。

返回：

Tensor。计算所得向量。shape为 $(...,D)$。

支持平台：

Ascend GPU