Batch Normalization

题面

实现 2D 输入张量的 BatchNorm 前向：输入 [N,C]，对每个特征 j 计算均值/方差并标准化，再用 gamma/beta 缩放平移（含 eps）。

Implementation Requirements

• Use only native features (external libraries are not permitted)
• The solve function signature must remain unchanged
• 输出写入 output

Examples

见页面两组示例（不同 gamma/beta）。

Constraints

• 1 ≤ N ≤ 10,000；1 ≤ C ≤ 1,024；eps=1e-5
• -100.0 ≤ input ≤ 100.0；0.1 ≤ gamma ≤ 10.0；-10.0 ≤ beta ≤ 10.0
• Performance: N = 5,000