Получить среднеквадратичную ошибку между каждым элементом в двух тензорах

У меня есть два одномерных тензора, y_pred и y_true куда:

>>> y_pred.shape
torch.Size([2730441, 1])
>>> y_true.shape
torch.Size([2730441, 1])

Чтобы получить среднеквадратичную ошибку между двумя тензорами, я могу использовать torch.nn.MSELoss() Однако я хочу получить потери между каждой строкой/элементом в тензорах y_pred и y_true, то есть я хочу запустить некоторую функцию elementWiseMSE(y_pred, y_true), которая вернет loss_tensor формы [2730441, 1], которая содержит среднюю поэлементную ошибку всех прогнозов.