python中,可以用动态for循环对不同维度的数据,比如图片、张量等,实现所有元素求和、范数求和。
进行检验:
验证结果:
一般Loss函数中,都是求权重L2范数的平方,比如:
此时,上面的具体操作处应该改为:
再次对dim4、dim5进行检验的结果为: