mindchemistry.e3.o3.matrix_z

mindchemistry.e3.o3.matrix_z(angle)

Give the rotation matrices around z axis for given angle.

Parameters

angle (Union[Tensor[float32], List[float], Tuple[float], ndarray[np.float32], float]) – The rotation angles around z axis. The shape of 'angle' is \((...)\).

Returns

Tensor, the rotation matrices around z axis. The shape of output is \((..., 3, 3)\).

Supported Platforms:

Ascend

Examples

>>> 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.       ]]