如何獲得特定維度上張量的 MSE？

Question

我有2張量與.size的torch.Size([2272, 161]) 我想得到它們之間的均方誤差。 不過，我想它沿着每個161個通道，使我的錯誤張量有.size的torch.Size([161]) 我怎樣才能做到這一點？

似乎torch.nn.MSELoss不允許我指定維度。

Answer 1

對於nn.MSELoss您可以指定選項reduction='none' 。 然后，這將為您返回兩個張量的每個條目位置的平方誤差。 然后你可以應用torch.sum/torch.mean。

a = torch.randn(2272,161)
b = torch.randn(2272,161)
loss = nn.MSELoss(reduction='none')
loss_result = torch.sum(loss(a,b),dim=0) 

我認為沒有一種直接的方法可以在損失的初始化時指定應用均值/總和的維度。 希望有幫助！