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,)