mindchemistry.e3.o3.matrix_z

mindchemistry.e3.o3.matrix_z(angle)

给出给定角度下绕z轴的旋转矩阵。

参数：

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

返回：

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

支持平台：

Ascend

样例：

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