Hi
I would like to use torch.nn.MSELoss() to compute the average root mean squared error (aRMSE). Therefore I want to take the root of the MSE and AFTER that average the error.
My problem is, when I apply the MSEloss() to predictions/references with multiple features and multiple samples, it still only returns a single value. Is it possible to us MSEloss() to only average over one dimension?
Regards