mindsponge.common.rots_from_tensor

mindsponge.common.rots_from_tensor(rots, use_numpy=False)[源代码]

输入tensor，将最后两根轴对应的3*3的旋转矩阵摊平拆分，得到旋转矩阵的每个分量，rots_to_tensor的逆操作。

参数：

rots (Tensor) - 代表旋转矩阵，shape为 \((..., 3, 3)\) 。
use_numpy (bool) - 是否使用numpy计算，默认值： False。

返回：

tuple，使用向量表示的旋转矩阵，矩阵为 \((xx, xy, xz, yx, yy, yz, zx, zy, zz)\) 。

支持平台：

Ascend GPU

样例：

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