BatchNorm 前向(训练模式)

大模型手撕 · 简单 · 归一化、基础算子

实现 BatchNorm1d 训练模式的前向:输入 x (N, C),对每个通道 c 用当前 batch 的均值和有偏方差归一化:

数据范围

在 RagCodes 上用 Python / C++ / Java 在线编写并提交,核心代码模式与 ACM 模式均可评测。