MSELoss定义:
其中n是batch_size,如果将reduce设置为true,则:
计算加法运算符后,仍将被n除。如果size_average设置为False,则避免将其除以n;
参数:
size_average (bool, optional):已废除。 缺省loss在计算输入batch的损失后计算平均值。 如果sample包含多个元素,则如果size_average设置为false,则loss将对每个batch求和;如果reduce设置为false,则参数设置为false,默认值为True。
reduce (bool, optional):默认值为True,目前已废除。 缺省情况下,loss将合计batch数据并计算平均值。 如果reduce设置为False,则表示size_average设置为False,仅合计batch的loss,不求平均。
reduction (string, optional):设置输出的计算方法。 有三个选项:无。 none )求出各要素的loss后,不进行追加计算; element wise _ mean :将loss的结果平均化; sum :只需合计所有的loss即可。 注:参数size_average和reduce已过时。 设置这两个参数之一将复盖reduction。 默认选项: elementwise_mean。
使用案例:
loss=nn.MSE loss (input=torch.randn ) 3,5,requires_grad=True ) target=torch.randn ) 3,5 ) output=looss