请按照初始代码中的签名实现 BatchNorm 类。
x 为 float32 (N, C),2 <= N <= 32,1 <= C <= 64;gamma/beta 为 (C,) float32;eps > 0 且通常为 1e-5。
单位缩放基础用例
正确性调用
BatchNorm(tensor[4] [1,1,1,1], tensor[4] [0,0,0,0], 0.00001).forward(tensor[6, 4] [-0.295519, 5.81772, 1.90989, 3.31025, 5.37078, 4.64916, …])
预期结果
tensor[6, 4] [-0.73817, 1.47546, -0.300254, 1.47894, 0.646824, 0.833124, …]
运行会检查可见示例,提交会运行完整的隐藏测试。
登录并提交后,你的提交记录会显示在这里。