mindscience.e3nn.o3.matrix_z

mindscience.e3nn.o3.matrix_z(angle)

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

参数：

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

返回：

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

样例：

>>> from mindscience.e3nn.o3 import matrix_z
>>> m = matrix_z(0.6)
>>> print(m)
[[ 0.8253357 -0.5646425  0.       ]
[ 0.5646425  0.8253357  0.       ]
[ 0.         0.         1.       ]]