python - 特定の次元でテンソルの MSE を取得するにはどうすればよいですか?

Question

の2 つのテンソルが.sizeありtorch.Size([2272, 161])ます。それらの間の平均二乗誤差を取得したい。ただし、161 チャネルのそれぞれに沿って、エラー テンソル.sizeがtorch.Size([161]). どうすればこれを達成できますか?

torch.nn.MSELoss寸法指定はできないようです。