mindsponge.common.rots_expand_dims

mindsponge.common.rots_expand_dims(rots, axis)[源代码]

对旋转矩阵 rots 的各个部分在指定的轴上添加额外维度。

参数：

rots (Tuple) - 旋转矩阵 $(x x, x y, x z, y x, y y, y z, z x, z y, z z)$ ，且xx, xy等均为Tensor且shape相同。
axis (Int) - 新插入的维度的位置，仅接受常量输入。

返回：

rots, Tuple。如果 axis 的值为0， xx 的 shape 为 $(. . ., X_{R})$ ,其中 $X_{R}$ 为任意数，拓展后为 $(1, . . ., X_{R})$ ,若axis不为0则在对应轴拓展，返回拓展后的 $(x x, x y, x z, y x, y y, y z, z x, z y, z z)$ 矩阵。

支持平台：

Ascend GPU

样例：

>>> from mindsponge.common.geometry import rots_expand_dims
>>> from mindspore.common import Tensor
>>> from mindspore import dtype as mstype
>>> rots = (1, 2, 3, 4, 5, 6, 7, 8, 9)
>>> axis = 0
>>> rots_expand_dims(rots, axis)
>>> print(output)
(Tensor(shape=[1], dtype=Int64, value=[1]), Tensor(shape=[1], dtype=Int64, value=[2]),
Tensor(shape=[1], dtype=Int64, value=[3]), Tensor(shape=[1], dtype=Int64, value=[4]),
Tensor(shape=[1], dtype=Int64, value=[5]), Tensor(shape=[1], dtype=Int64, value=[6]),
Tensor(shape=[1], dtype=Int64, value=[7]), Tensor(shape=[1], dtype=Int64, value=[8]),
Tensor(shape=[1], dtype=Int64, value=[9]))