mindscience.e3nn.o3.matrix_x

mindscience.e3nn.o3.matrix_x(angle)

返回绕 x 轴旋转给定角度的 \(3 \times 3\) 旋转矩阵。

参数：

• angle (Union[Tensor[float32], list[float], tuple[float], ndarray[np.float32], float]) - 围绕x轴的旋转角度。形状为 \((...)\)。

返回：

Tensor，绕 x 轴的旋转矩阵。输出形状为 \((..., 3, 3)\)。

样例：

>>> from mindscience.e3nn.o3 import matrix_x
>>> m = matrix_x(0.4)
>>> print(m)
[[ 1.          0.          0.        ]
[ 0.          0.92106086 -0.38941833]
[ 0.          0.38941833  0.92106086]]