Skip to content

core.base_abstractions.distributions#

[view_source]

CategoricalDistr#

class CategoricalDistr(Distr)

[view_source]

A categorical distribution extending PyTorch's Categorical.

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.