请按照下方说明实现 layer_norm()。
x 的形状为(..., d),其中 d ≥ 2;weight 和 bias 是形状为(d,)的张量;eps 是正浮点数。
归一化两行
正确性调用
layer_norm(tensor[2, 3] [1,2,3,4,5,6], tensor[3] [1,1,1], tensor[3] [0,0,0], eps=0.00001)
预期结果
tensor[2, 3] [-1.224736,0,1.224736,-1.224736,0,1.224736]
运行会检查可见示例,提交会运行完整的隐藏测试。
登录并提交后,你的提交记录会显示在这里。