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.distribution.ProbabilityDistribution` interface
          
docstr_tok: ["implementation","of",":","class",":","`","~maze.distributions.distribution.ProbabilityDistribution","`","interface"]
code: def deterministic_sample(self):
      """implementation of :class:`~maze.distributions.distribution.ProbabilityDistribution` interface"""
      return self.logits.argmax(dim=-1)
  
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 4 for calling the CategoricalProbabilityDistribution obj's underlying member method code with example usage: obj.deterministic_sample() and returns: self