mindsponge.common.quat_to_rot

mindsponge.common.quat_to_rot(normalized_quat, use_numpy=False)

将四元数转化为旋转矩阵。

$$\begin{array}{r}\begin{array}{rl}& xx=1-2\ast y\ast y-2\ast z\ast z\\ & xy=2\ast x\ast y+2\ast w\ast z\\ & xz=2\ast x\ast z-2\ast w\ast y\\ & yx=2\ast x\ast y-2\ast w\ast z\\ & yy=1-2\ast x\ast x-2\ast z\ast z\\ & yz=2\ast z\ast y+2\ast w\ast x\\ & zx=2\ast x\ast z+2\ast w\ast y\\ & zy=2\ast y\ast z-2\ast w\ast x\\ & zz=1-2\ast x\ast x-2\ast y\ast y\end{array}\end{array}$$

参数：

• normalized_quat (tensor) - 归一化的四元数，shape为 $(N_{res},4)$ 。

• use_numpy (bool) - 是否使用numpy计算，默认值： False。

返回：

旋转矩阵(tuple)， $(xx,xy,xz,yx,yy,yz,zx,zy,zz)$ , 每个元素shape $(N_{res},)$ 。

支持平台：

Ascend GPU

样例：

>>> import numpy as np
>>> import mindspore as ms
>>> from mindspore import Tensor
>>> from mindsponge.common.geometry import quat_to_rot
>>> input_0 = Tensor(np.ones((256, 4)), ms.float32)
>>> output = quat_to_rot(input_0)
>>> print(len(output), output[0].shape)
9, (256,)