sccross.models.layers.Prior

class sccross.models.layers.Prior(loc=0.0, std=1.0)[source]

Bases: torch.nn.modules.module.Module

Prior distribution

Parameters

  • loc (float) – Mean of the normal distribution

  • std (float) – Standard deviation of the normal distribution

Methods

forward

Define the computation performed at every call.

Attributes