pytorch api torch.nn.MSELoss

版权声明：本文为作者创作，转载请注明出处:http://blog.csdn.net/claroja，如有商业用途请联系QQ：63183535。 https://blog.csdn.net/claroja/article/details/85327171

CLASS torch.nn.MSELoss(size_average=None, reduce=None, reduction='mean')
计算输入X和输出Y的均方误差.

参数	描述
reduction	(string,optional),默认为平均值

loss = nn.MSELoss()
input = torch.randn(3, 5, requires_grad=True)
target = torch.randn(3, 5)
output = loss(input, target)

参考文献:
https://pytorch.org/docs/stable/nn.html#torch.nn.MSELoss