sponge.function.calc_torsion_by_vectors

查看源文件
sponge.function.calc_torsion_by_vectors(vector1: Tensor, vector2: Tensor, axis_vector: Tensor = None, keepdims: bool = False)[源代码]

计算由三个向量形成的扭转角。

参数:
  • vector1 (Tensor) - 向量1,shape为 \((..., D)\) ,D是模拟系统的维度。

  • vector2 (Tensor) - 向量2,shape为 \((..., D)\)

  • axis_vector (Tensor) - 轴向量,shape为 \((..., D)\)

  • keepdims (bool) - 设置为 True 的话,最后一个维度会保留,默认值 False

返回:

Tensor。计算所得扭转角。shape为 \((...)\)\((..., 1)\)

支持平台:

Ascend GPU