Action Spaces and Distributions Module

This page contains the reference documentation of Maze Action Spaces and Distributions Module.

These are interfaces, classes and utility functions:

ProbabilityDistribution	Base class for all probability distributions.
TorchProbabilityDistribution	Base class for wrapping Torch probability distributions.
DistributionMapper	Provides a mapping of spaces and action heads to the respective probability distributions to be used.
atanh	Computes the arc-tangent hyperbolic.
tensor_clamp	Clamping with tensor and broadcast support.

These are built-in Torch probability distributions:

CategoricalProbabilityDistribution	Categorical Torch probability distribution.
BernoulliProbabilityDistribution	Bernoulli Torch probability distribution for multi-binary action spaces.
DiagonalGaussianProbabilityDistribution	Diagonal Gaussian (Normal) Torch probability distribution.
SquashedGaussianProbabilityDistribution	Tanh-squashed diagonal Gaussian (Normal) Torch probability distribution.
BetaProbabilityDistribution	Beta Torch probability distribution.

These are combined probability distributions:

MultiCategoricalProbabilityDistribution	Multi-categorical probability distribution.
DictProbabilityDistribution	Dictionary probability distribution.