mindscience.e3nn.o3.xyz_to_angles

mindscience.e3nn.o3.xyz_to_angles(xyz)[源代码]

将单位球面上的点 \(\vec r = (x, y, z)\) 转换为角度 \((\alpha, \beta)\)

\[\vec r = R(\alpha, \beta, 0) \vec e_z\]
参数:
  • xyz (Tensor) - 点 \((x, y, z)\)。形状为 \((..., 3)\) 的张量。

返回:

tuple[Tensor],由 \(\alpha\)\(\beta\) 组成的二元组。

样例:

>>> import mindspore as ms
>>> from mindscience.e3nn.o3 import xyz_to_angles
>>> input = ms.Tensor([3, 3, 3])
>>> m = xyz_to_angles(input)
>>> print(m)
(Tensor(shape=[], dtype=Float32, value= 0.785398), Tensor(shape=[], dtype=Float32, value= 0.955318))