3.1.22.11. unit_scaling.functional.mse_loss

unit_scaling.functional.mse_loss(input: Tensor, target: Tensor, size_average: bool | None = None, reduce: bool | None = None, reduction: str = 'mean') → Tensor[source]

Computes the unit-scaled element-wise mean squared error.

Measures the element-wise mean squared error. See MSELoss for details.