cyto_dl.nn.losses.weighted_mse_loss module

class cyto_dl.nn.losses.weighted_mse_loss.WeightedMSELoss(reduction='none', weights=1)

Bases: _Loss

forward(input: Tensor, target: Tensor) → Tensor