UnsqueezeV2

产品支持情况

产品	是否支持
Ascend 950PR&950DT系列产品	√
Atlas A3系列产品	√
Atlas A2系列产品	√
Atlas 200I/500 A2推理产品	√
Atlas推理系列产品	√
Atlas训练系列产品	√

功能说明

算子功能：在Tensor的shape中插入维度值为1的维度。axis以属性的形式提供。该算子为已废弃版本，请优先使用Unsqueeze算子。示例：

axis指定时，在指定位置插入维度值为1的维度：

>>> x = torch.randn(2, 3)
>>> torch.unsqueeze(x, 1).shape
(2, 1, 3)

axis支持负数索引，-1表示在末尾插入：

>>> x = torch.randn(2, 3)
>>> torch.unsqueeze(x, -1).shape
(2, 3, 1)

axis为空或不指定时，不插入任何维度，输出shape与输入shape一致。

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入	待扩维的输入张量	所有类型	ND
axis	属性	待插入维度的位置列表，取值范围为[-rank(x)-len(axis), rank(x)+len(axis))，支持负数索引 位置不能重复（按负数索引归一化后的位置判定） 不指定或为空时，不插入任何维度，输出shape与输入shape一致	ListInt，默认值为[]	-
y	输出	扩维后的输出张量，数据类型与输入x一致	所有类型	ND

约束说明

• axis取值超出[-rank(x)-len(axis), rank(x)+len(axis))范围，或归一化后位置重复时，算子执行失败。
• 输出维数rank(x)+len(axis)超出框架最大维数限制时，算子执行失败。
• axis为空或不指定时，输出shape与输入shape一致。
• 输入shape为-2时，输出shape为-2。
• 输入含-1（未知维）时，未知维原样保留。

调用说明

调用方式	样例代码	说明
图模式调用	test_geir_unsqueezev2	通过算子IR构图方式调用UnsqueezeV2算子。