如果loss有多个维度会怎样
收藏
loss是一个tensor,tensor一个维度可以有多个值,也可以有多个维度,我看提供的都是一维的一个值,比如[5.33],但是tensor也可以是[5.33, 1.22],如果loss是这样的值,会怎样
0
收藏
请登录后评论
分别对每个loss求梯度,合并平均后返回