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