harmony 鸿蒙Squeeze

2023-06-24 浏览 (631)

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。

输入:

  • x,n维tensor

输出:

  • output,去除axis中长度为1的维度之后得到的tensor。

Since:

3.2

Version:

1.0

相关模块:

NNRt

汇总

Public 属性

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

类成员变量说明

axis

long [] Squeeze::axis

描述:

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

你可能感兴趣的鸿蒙文章

harmony 鸿蒙HDI 参数参考

harmony 鸿蒙ActRecognitionEvent

harmony 鸿蒙Activation

harmony 鸿蒙ActivityRecognitionTypes.idl

harmony 鸿蒙AddFusion

harmony 鸿蒙Alignment

harmony 鸿蒙AllocInfo

harmony 鸿蒙ApduSimIORequestInfo

harmony 鸿蒙ArgMaxFusion

harmony 鸿蒙Audio

  • 所属分类: 后端技术
  • 本文标签: 鸿蒙 软件
  • 版权声明: 本文链接 https://seaxiang.com/blog/e396c9fe24304ad29ff66362031d23a4