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","."]
docstr: implementation of :class:`~maze.distributions.torch_dist.TorchProbabilityDistribution` interface
          
docstr_tok: ["implementation","of",":","class",":","`","~maze.distributions.torch_dist.TorchProbabilityDistribution","`","interface"]
code: def required_logits_shape(
      cls, action_space: spaces.Discrete
  ) -> Sequence[int]:
      """implementation of :class:`~maze.distributions.torch_dist.TorchProbabilityDistribution` interface"""
      return [action_space.n]
  
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 1 for calling the CategoricalProbabilityDistribution obj's underlying member method code with example usage: obj.required_logits_shape(cls, action_space) and returns: list