CANN/ops-nn 分组归一化梯度算子-北京尧图网络

GroupNormalizationGrad【免费下载链接】ops-nn本项目是CANN提供的神经网络类计算算子库实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-nn产品支持情况产品是否支持Atlas A2 训练系列产品/Atlas 800I A2 推理产品√功能说明算子功能计算 Group Normalization 的输入梯度。计算公式输入按[N, G, M]布局组织单 group 的计算公式为$$ \hat{x} (x - mean) \cdot rstd $$$$ s_1 \sum(dy \cdot gamma), \quad s_2 \sum(dy \cdot gamma \cdot \hat{x}) $$$$ dx \frac{rstd}{M} \cdot gamma \cdot (M \cdot dy - s_1 - \hat{x} \cdot s_2) $$其中 $x$ 为前向输入$dy$ 为上游梯度$gamma$ 为已广播到[N, G, M]的缩放系数$mean$ 为每个 group 的均值形状[N, G]$rstd$ 为每个 group 的标准差倒数形状[N, G]$dx$ 为输出输入梯度形状与 $x$ 相同。参数说明参数名输入/输出/属性描述数据类型数据格式x输入前向输入公式中的 x。FLOAT、FLOAT16、BFLOAT16NDdy输入上游梯度公式中的 dy。FLOAT、FLOAT16、BFLOAT16NDgamma输入已广播到 [N, G, M] 的缩放系数。FLOAT、FLOAT16、BFLOAT16NDmean输入每个 group 的均值形状为 [N, G]。FLOAT、FLOAT16、BFLOAT16NDrstd输入每个 group 的标准差倒数形状为 [N, G]。FLOAT、FLOAT16、BFLOAT16NDdx输出输入梯度公式中的 dx。FLOAT、FLOAT16、BFLOAT16ND约束说明输入 x、dy、gamma 和输出 dx 的数据类型需保持一致。输入 x、dy、gamma 的 shape 需相同。输入维度至少为 3 维[N, G, M] 布局。mean 和 rstd 需为 [N, G] 形状。贡献说明贡献者贡献方贡献算子贡献时间贡献内容Zhou Jianhua LePenseurAISS Group, Harbin Institute of Technology (HIT)GroupNormalizationGrad2026/05/12GroupNormalizationGrad 算子架构迁移至标准 AscendC 框架【免费下载链接】ops-nn本项目是CANN提供的神经网络类计算算子库实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-nn创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考

CANN/ops-nn 分组归一化梯度算子

相关推荐

Grape-Entity 安全实践：保护 API 数据暴露的 5 个关键点

如何在5分钟内上手Ddisasm？从安装到生成GTIRB文件的完整指南

Ornith-1.0-35B-OptiQ-4bit快速上手指南：3步完成本地部署，支持图像文本输入的强大AI助手

更多资讯

VOS3000操作手册精讲：费率套餐配置与话单报表避坑指南

信息学竞赛校级预选赛：变量命名、赋值语法与数组索引三大基础考点解析

知识竞赛系统面试高频考点：幂等设计、实时榜单与高并发防作弊

MATLAB矩阵对角化实战：特征值、若当标准形与符号运算避坑指南

大数运算实现详解：从数组存储到快速幂的完整设计

Java Swing 麦田物语课程设计：从零构建种田交易与存档的完整指南

今日更新

MATLAB缩进快捷键与批量格式化实战指南

用Poco实现Qt风格信号与槽的跨平台C++方案

ESXi裸机虚拟化安装指南：从硬件兼容到虚拟机部署

本周精选

金融AI落地指南：从调研报告到技术选型与工程实践

大模型上下文成本与注意力优化技术深度解析——为何O(n²)平方复杂度无法被常规优化消除｜TaoToken 统一 Key 通道实测

基于GMM的语音识别系统原理与MATLAB实现详解

本月热门

我发现了一个新思路：用 Remotion + Claude Code 像写代码一样自动化生成短视频

Windows下 Codex 中 Chrome 和 Computer Use 插件不可用问题排查及解决参考方式：TaoToken 统一 Key 配置与验证

2026 大模型集体涨价：用 Python 做企业 Token 成本测算与选型避坑（附配置）

读完文章，想聊聊您的网站？