sccross.models.utils.MSE

class sccross.models.utils.MSE(loc)[source]

Bases: torch.distributions.distribution.Distribution

A “sham” distribution that outputs negative MSE on log_prob

Parameters

loc (Tensor) – Mean of the distribution

Methods

log_prob

Returns the log of the probability density/mass function evaluated at value.

Attributes

mean

Returns the mean of the distribution.