昇腾社区首页
中文
注册

Squeeze

功能

从输入中去除尺寸为1的维度。

输入

一个输入:

x:tensor,数据类型:float16、float、double、uint8、uint16、uint32、uint64、int8、int16、int32、int64、bool。

输出

一个输出:

y:tensor,数据类型和输入一致。

属性

axes:一个数据类型为int32或者int64的整形列表,维度为1;取值范围为[-r, r-1](r表示输入tensor的秩,负数表示从后面计算维度);含义:指定要去除的维度。

支持的ONNX版本

Opset v8/v9/v10/v11/v12/v13/v14/v15/v16/v17/v18