Skip to content

core.base_abstractions.distributions#

[view_source]

CategoricalDistr#

class CategoricalDistr(Distr)

[view_source]

A categorical distribution extending PyTorch's Categorical.

FixedNormal#

class FixedNormal(torch.distributions.Normal)

[view_source]

A fixed normal distribution extending PyTorch's Normal.

FixedBernoulli#

class FixedBernoulli(torch.distributions.Bernoulli)

[view_source]

A fixed Bernoulli distribution extending PyTorch's Bernoulli.

DiagGaussian#

class DiagGaussian(nn.Module)

[view_source]

A learned diagonal Gaussian distribution.

Bernoulli#

class Bernoulli(nn.Module)

[view_source]

A learned Bernoulli distribution.

AddBias#

class AddBias(nn.Module)

[view_source]

Adding bias parameters to input values.

AddBias.__init__#

 | __init__(bias: torch.FloatTensor)

[view_source]

Initializer.

Parameters

  • bias : data to use as the initial values of the bias.

AddBias.forward#

 | forward(x: torch.FloatTensor) -> torch.FloatTensor

[view_source]

Adds the stored bias parameters to x.