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