iM ddlmZddlZddlmZddlmZdedededeeeffd Zdd ed eeefd e defd Z ddeded e dedef dZ y))UnionN)Tensor)_check_same_shapepredstarget num_outputsreturnct|||dk(r"|jd}|jd}||z }tj||zd}||jdfS)aUpdate and returns variables required to compute Mean Squared Error. Check for same shape of input tensors. Args: preds: Predicted tensor target: Ground truth tensor num_outputs: Number of outputs in multioutput setting r)dim)rviewtorchsumshape)rrrdiffsum_squared_errors {/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/torchmetrics/functional/regression/mse.py_mean_squared_error_updatersaeV$a 2R 6>D $+15 fll1o --rnum_obssquaredc@|r||z Stj||z S)a Compute Mean Squared Error. Args: sum_squared_error: Sum of square of errors over all observations num_obs: Number of predictions or observations squared: Returns RMSE value if set to False. Example: >>> preds = torch.tensor([0., 1, 2, 3]) >>> target = torch.tensor([0., 1, 2, 2]) >>> sum_squared_error, num_obs = _mean_squared_error_update(preds, target, num_outputs=1) >>> _mean_squared_error_compute(sum_squared_error, num_obs) tensor(0.2500) )rsqrt)rrrs r_mean_squared_error_computer*s' +2 w &^uzzBSV]B]7^^rc@t|||\}}t|||S)aCompute mean squared error. Args: preds: estimated labels target: ground truth labels squared: returns RMSE value if set to False num_outputs: Number of outputs in multioutput setting Return: Tensor with MSE Example: >>> from torchmetrics.functional.regression import mean_squared_error >>> x = torch.tensor([0., 1, 2, 3]) >>> y = torch.tensor([0., 1, 2, 2]) >>> mean_squared_error(x, y) tensor(0.2500) )r)r)rr)rrrrrrs rmean_squared_errorr=s)("r$s ;.f.f.3.SXY_adYdSe.(_6_E#v+DV_ae_qw_&TfTfTtTY\TekTr