mindsponge.common.vecs_expand_dims

mindsponge.common.vecs_expand_dims(v, axis)[源代码]

将输入的v在指定的轴添加额外维度。

参数：

v (Tuple) - 输入的初始向量，长度为3，\((xx, xy, xz)\)。
axis (int) - 新插入的维度的位置,仅接受常量输入。

返回：

tuple，如果 axis 的值为0，且 \(xx\) 的shape为 \((... , X_R)\) ，拓展后的shape为 \((1, ..., X_R)\) 。若 axis为其它值，则在其它方向拓展，返回拓展后的 \((xx, xy, xz, yx, yy, yz, zx, zy, zz)\) 。

支持平台：

Ascend GPU

样例：

>>> from mindsponge.common.geometry import vecs_expand_dims
>>> from mindspore.common import Tensor
>>> from mindspore import dtype as mstype
>>> v = (1, 2, 3)
>>> axis = 0
>>> output= vecs_expand_dims(v, axis)
>>> print(output)
(Tensor(shape=[1], dtype=Int64, value=[1]),Tensor(shape=[1], dtype=Int64, value=[2]),
Tensor(shape=[1], dtype=Int64, value=[3]))