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))