CategoricalProbabilityDistribution¶
-
class
maze.distributions.categorical.
CategoricalProbabilityDistribution
(*args, **kwds)¶ Categorical Torch probability distribution.
- Parameters
logits – the action selection logits.
-
deterministic_sample
()¶ implementation of
ProbabilityDistribution
interface
-
log_prob
(actions: torch.Tensor) → torch.Tensor¶ implementation of
ProbabilityDistribution
interface
-
classmethod
required_logits_shape
(action_space: gym.spaces.Discrete) → Sequence[int]¶ implementation of
TorchProbabilityDistribution
interface