文件最后提交记录最后更新时间
20 天前
20 天前
15 天前
20 天前
10 个月前
20 天前
5 天前
README

UnsqueezeV3

产品支持情况

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

功能说明

接口功能:在输入Tensor的指定位置插入维度值为1的维度,仅改变shape不改变数据。axes以输入的形式提供。示例:

axes指定时,在对应位置插入维度1:

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

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
x 输入 待插入维度的输入张量 所有类型 ND
axes 输入
  • 待插入维度的位置列表,取值范围为[-(rank(x)+len(axes)), rank(x)+len(axes))
  • 列表中的值不能重复
INT64 ND
y 输出 插入维度后的输出张量,数据类型与输入x一致 所有类型 ND

约束说明

  • axes取值超出[-(rank(x)+len(axes)), rank(x)+len(axes))范围时,算子执行失败。
  • axes包含重复值时,算子执行失败。
  • axes为空时,输出shape与输入x一致。
  • 编译阶段无法获取axes数据(如axes来自上游计算结果)时,输出shape置为-2。
  • 输入shape为-2时,输出shape为-2。

调用说明

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