core.base_abstractions.distributions
#
CategoricalDistr
#
class CategoricalDistr(Distr)
A categorical distribution extending PyTorch's Categorical.
AddBias
#
class AddBias(nn.Module)
Adding bias parameters to input values.
AddBias.__init__
#
| __init__(bias: torch.FloatTensor)
Initializer.
Parameters
- bias : data to use as the initial values of the bias.
AddBias.forward
#
| forward(x: torch.FloatTensor) -> torch.FloatTensor
Adds the stored bias parameters to x
.