harmony 鸿蒙AvgPoolFusion

  • 2025-06-16
  • 浏览 (10)

AvgPoolFusion

概述

在输入tensor上应用 2D 平均池化。支持int8量化输入。

该算子对应的NodeType为NODE_TYPE_AVGPOOL_FUSION。

参数限制:当padMode==PAD_MODE_PAD时,padList的数值必须大于等于0。其他情况下padding的数值必须为0。

输入:

  • x,n维张量。

输出:

  • output, 输出平均池化后的张量。

起始版本: 3.2

相关模块:NNRt

汇总

Public 属性

名称 描述
long[] kernelSize 用来取平均值的kernel大小,是一个长度为2的int数组[kernel_height,kernel_weight], 第一个数表示kernel高度,第二个数表示kernel宽度。
long[] strides kernel移动的距离,是一个长度为2的int数组[stride_height,stride_weight], 第一个数表示高度上的移动步幅,第二个数表示宽度上的移动步幅。
long[] pad x周围的填充,是一个长度为4的int数组[top,bottom,left,right],并且以最近邻的值填充。
enum PadModepadMode 填充模式
enum RoundModeroundMode 取整数的算法
enum Formatformat 运算时的数据排列排列,详情请参考:Format
boolean global 是否是全局池化
enum ActivationTypeactivationType 激活函数,详情请参考:ActivationType

类成员变量说明

activationType

enum ActivationType AvgPoolFusion::activationType

描述

激活函数,详情请参考:ActivationType

format

enum Format AvgPoolFusion::format

描述

运算时的数据排列排列,详情请参考:Format

global

boolean AvgPoolFusion::global

描述

是否是全局池化

kernelSize

long [] AvgPoolFusion::kernelSize

描述

用来取平均值的kernel大小,是一个长度为2的int数组[kernel_height,kernel_weight], 第一个数表示kernel高度,第二个数表示kernel宽度。

pad

long [] AvgPoolFusion::pad

描述

x周围的填充,是一个长度为4的int数组[top,bottom,left,right],并且以最近邻的值填充。

padMode

enum PadMode AvgPoolFusion::padMode

描述

填充模式

roundMode

enum RoundMode AvgPoolFusion::roundMode

描述

取整数的算法

strides

long [] AvgPoolFusion::strides

描述

kernel移动的距离,是一个长度为2的int数组[stride_height,stride_weight], 第一个数表示高度上的移动步幅,第二个数表示宽度上的移动步幅。

你可能感兴趣的鸿蒙文章

harmony 鸿蒙NNRt HDI 参考

harmony 鸿蒙Activation

harmony 鸿蒙Activation

harmony 鸿蒙AddFusion

harmony 鸿蒙AddFusion

harmony 鸿蒙ArgMaxFusion

harmony 鸿蒙ArgMaxFusion

harmony 鸿蒙AvgPoolFusion

harmony 鸿蒙BatchToSpaceND

harmony 鸿蒙BatchToSpaceND

0  赞