请按照初始代码中的签名实现 Conv2D 类。
x 为 float32 (B, C_in, H, W),H, W <= 16;weight 为 (C_out, C_in, K, K),K ∈ {1, 3, 5};bias 为 (C_out,);stride ∈ {1, 2};padding ∈ {0, 1, 2};B, C_in, C_out <= 4。
单通道 3x3 无填充
正确性调用
Conv2D(tensor[2, 1, 3, 3] [1.69831, -0.650177, 0.649779, -0.855563, 0.487731, 0.77782, …], tensor[2] [-0.132222,-0.653882], 1, 0).forward(tensor[1, 1, 5, 5] [0.340566, -0.474313, -0.178415, -0.468288, -0.855166, 0.910877, …])
预期结果
tensor[1, 2, 3, 3] [0.511498, -0.786528, -3.19211, 2.5726, -5.20519, 1.7238, …]
运行会检查可见示例,提交会运行完整的隐藏测试。
登录并提交后,你的提交记录会显示在这里。