distribution

Distribution subpackage.

Submodules

array_bernoulli

NumPy-based Bernoulli distribution representation.

array_categorical

Numpy-based distribution representation.

array_dirichlet

Numpy-based distribution representation.

array_gaussian

Numpy-based Gaussian distribution representation.

torch_bernoulli

Torch-based Bernoulli distribution representation.

torch_categorical

Torch-based categorical distribution representation.

torch_dirichlet

Torch implementation of the Dirichlet distribution representation.

torch_gaussian

Torch-based Gaussian distribution representation.

torch_mixture

Torch-based mixture distribution representation.

torch_sparse_log_categorical

Sparse torch categorical distributions represented by grouped logits.