CANN: 1D平均池化平均矩阵算子

AvgPool1DAvgMatrix【免费下载链接】ops-nn本项目是CANN提供的神经网络类计算算子库实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-nn产品支持情况产品是否支持Ascend 950PR/Ascend 950DT√Atlas A3 训练系列产品/Atlas A3 推理系列产品√Atlas A2 训练系列产品/Atlas A2 推理系列产品√Atlas 200I/500 A2 推理产品×Atlas 推理系列产品√Atlas 训练系列产品√功能说明算子功能为1DAvgPool反向传播场景生成平均矩阵权重。参数说明参数名输入/输出/属性描述数据类型数据格式x输入输入张量仅使用shape和format信息。FLOAT、FLOAT16、INT8、INT16、UINT8、INT32、INT64、DOUBLENCHW、NHWC、NC1HWC0ksize属性池化窗口大小。INT-strides属性滑动步长。INT-pads属性左右padding。ListInt-ceil_mode属性是否使用向上取整。默认false。BOOL-count_include_pad属性计数是否包含padding。默认false。BOOL-y输出平均矩阵权重输出。与x相同支持FLOAT、FLOAT16、INT8、INT16、UINT8、INT32、INT64、DOUBLENC1HWC0约束说明pads至少包含两个元素。strides不能为0。输入format仅支持NCHW、NHWC、NC1HWC0。调用说明调用方式调用样例说明图模式调用test_geir_avg_pool1d_avg_matrix通过算子IR构图方式调用AvgPool1DAvgMatrix算子。【免费下载链接】ops-nn本项目是CANN提供的神经网络类计算算子库实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-nn创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考