probly.representation.distribution.torch_dirichlet

Torch implementation of the Dirichlet distribution representation.

Classes

TorchDirichletDistribution

A Dirichlet distribution stored as a torch tensor.