У меня есть 2 тензора с .size из torch.Size([2272, 161]) . Я хочу получить среднеквадратичную ошибку между ними. Однако я хочу, чтобы это было по каждому из 161 канала, чтобы мой тензор ошибок имел .size из torch.Size([161]) . Как я могу это...