1.损失函数
(1)nn.MSELoss()
有三个可选参数:reduce、size_average、reduction
-
如果 reduce = False,那么 size_average 参数失效,直接返回向量形式的 loss
-
如果 reduce = True,那么 loss 返回的是标量
a)如果 size_average = True,返回 loss.mean(),即loss的平均值
b)如果 size_average = False,返回 loss.sum(),loss的和
-
reduction = ‘none’,直接返回向量形式的 loss
-
reduction = ‘sum’,返回loss之和
-
reduction = ''elementwise_mean,返回loss的平均值
-
reduction = ''mean,返回loss的平均值