• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1# Squeeze
2
3## 概述
4
5去除axis中,长度为1的维度。支持int8量化输入。
6
7假设输入的x的shape为[2,1,1,2,2],axis为[0,1],则输出的output的shape为[2,1,2,2],意思是第0维到第1维之间,长度为1的维度被去除。
8
9该算子对应的[NodeType](_n_n_rt_v20.md#nodetype)为NODE_TYPE_SQUARED_DIFFERENCE。
10
11输入:
12
13- x,n维tensor。
14
15输出:
16
17- output,去除axis中长度为1的维度之后得到的tensor。
18
19**起始版本:** 3.2
20
21**相关模块:**[NNRt](_n_n_rt_v20.md)
22
23## 汇总
24
25### Public 属性
26
27| 名称 | 描述 |
28| -------- | -------- |
29| long[] [axis](#axis) | 指定删除的维度。axis可以是一个整数或数组,整数的取值范围为[-n,n)。  |
30
31## 类成员变量说明
32
33### axis
34
35```
36long [] Squeeze::axis
37```
38**描述**
39
40指定删除的维度。axis可以是一个整数或数组,整数的取值范围为[-n,n)。
41