PyTorch の平均二乗誤差の計算方法

Question

torch.nn.MSELoss()
が、実際の計算と合いません。

import torch
a = torch.tensor([0.5, 0.5])
b = torch.tensor([0.0, -0.2])

mse_loss = torch.nn.MSELoss() 
print(mse_loss(a, b))

# 出力
# tensor(0.3700) 

実際の計算は、

import math
print(math.sqrt((0.5-0.0)**2 + (0.5-(-0.2))**2 ) / 2)
# 0.4301162633521313

となります。

Answer 1

MSELossはMean Squared Errorですからルートが不要です。

Comments

[teefpc]

ありがございました！