netmap.model.standardautoencoder.MSELoss

class netmap.model.standardautoencoder.MSELoss(*args: Any, **kwargs: Any)

Bases: Module

__init__(scale_factor=1.0)

Mean Squared Error Loss :type scale_factor: float :param scale_factor: Scale factor applied to predictions. :type scale_factor: float

forward(y_true, y_pred)

Compute the MSE loss. :type y_true: torch.Tensor :param y_true: Ground truth values. :type y_true: torch.Tensor :type y_pred: torch.Tensor :param y_pred: Predicted mean values. :type y_pred: torch.Tensor

Returns:

Mean Squared Error.

Return type:

torch.Tensor