SqueezeV3
产品支持情况
| 产品 |
是否支持 |
| Ascend 950PR&950DT系列产品 |
√ |
| Atlas A3系列产品 |
√ |
| Atlas A2系列产品 |
√ |
| Atlas 200I/500 A2推理产品 |
√ |
| Atlas推理系列产品 |
√ |
| Atlas训练系列产品 |
√ |
功能说明
接口功能：删除Tensor中维度值为1的维度。axes以可选输入的形式提供。示例：
axes为空时，删除所有维度值为1的维度：
>>> x = torch.randn(2, 1, 3)
>>> torch.squeeze(x).shape
(2, 3)
axes指定时，仅删除指定的维度值为1的维度：
>>> x = torch.randn(2, 1, 3)
>>> torch.squeeze(x, 1).shape
(2, 3)
参数说明
| 参数名 |
输入/输出/属性 |
描述 |
数据类型 |
数据格式 |
| x |
输入 |
待压缩的输入张量 |
所有类型 |
ND |
| axes |
可选输入 |
- 待删除的维度列表，取值范围为[-rank(x), rank(x))
- 不指定或为空时，删除所有维度值为1的维度
- 指定时，仅删除指定的维度，指定的维度值必须为1
|
INT64 |
ND |
| y |
输出 |
压缩后的输出张量，数据类型与输入x一致 |
所有类型 |
ND |
约束说明
- 指定axes时，axes对应维度的维度值必须为1或-1，否则算子执行失败。
- axes取值超出[-rank(x), rank(x))范围时，算子执行失败。
- 输入为动态shape（含-1）且axes为空时，无法确定-1维度是否为1，输出shape置为-2。
- 输入shape为-2时，输出shape为-2。
- 输入shape确定但axes在编译阶段无法获取（如axes来自上游计算结果）时，输出shape置为-2。
调用说明