harmony 鸿蒙Squeeze

Squeeze

去除axis中，长度为1的维度。支持int8量化输入。

假设输入的x的shape为[2，1，1，2，2]，axis为[0,1]，则输出的output的shape为[2，1，2，2]，意思是第0维到第1维之间，长度为1的维度被去除。

该算子对应的NodeType为NODE_TYPE_SQUARED_DIFFERENCE。

输入：

输出：

Since:

3.2

Version:

1.0

相关模块:

NNRt

名称	描述
axis	long[] 指定删除的维度。axis可以是一个整数或数组，整数的取值范围为[-n，n)。

long [] Squeeze::axis

描述:

指定删除的维度。axis可以是一个整数或数组，整数的取值范围为[-n，n)。

0 赞

所属分类： 后端技术
本文标签：
版权声明： 原创文章如转载，请注明本文链接: https://m.seaxiang.com/blog/e396c9fe24304ad29ff66362031d23a4