sponge.function.calc_torsion_by_vectors

View Source On Gitee
sponge.function.calc_torsion_by_vectors(vector1: Tensor, vector2: Tensor, axis_vector: Tensor = None, keepdims: bool = False)[source]
Compute torsion angle formed by two direction vectors v1 and v2

and an axis vector vaxis.

Parameters
  • vector1 (Tensor) – Tensor of shape (...,D). Data type is float. D is spatial dimension of the simulation system. Usually is 3. Direction vector v1

  • vector2 (Tensor) – Tensor of shape (...,D). Data type is float. Direction vector v2

  • axis_vector (Tensor) – Tensor of shape (...,D). Data type is float. Axis vector vaxis. Default: None.

  • keepdims (bool) – If this is set to True, the last axis will be left in the result as dimensions with size one. Default: False.

Returns

torsion (Tensor), a tensor of shape (...) or (...,1). Data type is float.

Supported Platforms:

Ascend GPU