mindchemistry.e3.o3.xyz_to_angles

查看源文件
mindchemistry.e3.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)\) 的张量。

返回:
  • alpha (Tensor) - alpha 欧拉角。形状为 \((...)\) 的张量。

  • beta (Tensor) - beta 欧拉角。形状为 \((...)\) 的张量。

支持平台:

Ascend

样例:

>>> import mindspore as ms
>>> from mindchemistry.e3.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))