Я пытаюсь вычислить среднюю ошибку двух тензоров горелки.Я хотел бы игнорировать / маскировать строки, где метки 0 (пропущенные значения).Как я могу изменить эту строку, чтобы учесть это ограничение?
torch.sqrt(((preds.detach() - labels) ** 2).mean()).item()
Заранее спасибо.