mindscience.e3nn.o3.matrix_y

mindscience.e3nn.o3.matrix_y(angle)

Return the \(3 \times 3\) rotation matrix for a rotation about the y-axis by the given angle.

Parameters

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

Returns

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

Examples

>>> from mindscience.e3nn.o3 import matrix_y
>>> m = matrix_y(0.5)
>>> print(m)
[[ 0.87758255  0.          0.47942555]
[ 0.          1.          0.        ]
[-0.47942555  0.          0.87758255]]