class_docstr: Categorical Torch probability distribution.
  
  :param logits: the action selection logits.
class_docstr_tok: ["Categorical","Torch","probability","distribution",".",":","param","logits",":","the","action","selection","logits","."]
code: def __init__(
      self,
      logits: torch.Tensor,
      action_space: spaces.Discrete,
      temperature: float,
  ):
      self.logits = logits / temperature
      super().__init__(
          dist=Categorical(logits=self.logits),
          action_space=action_space,
      )
  
file_path: maze/maze/distributions/categorical.py
imports: from typing import Sequence
  import torch
  from gym import spaces
  from torch.distributions import Categorical
  from maze.core.annotations import override
  from maze.distributions.torch_dist import TorchProbabilityDistribution
label_desc: Use image node_id 2 to create a new CategoricalProbabilityDistribution object from inherited base classes: TorchProbabilityDistribution with example: obj = CategoricalProbabilityDistribution(logits, action_space, temperature)