code: def __init__(
      self,
      distributional_network: ModelBase,
      *,
      qmin: float,
      qmax: float,
      num_atoms: int,
  ) -> None:
      super().__init__()
      self.distributional_network = distributional_network
      self.support = torch.linspace(qmin, qmax, num_atoms)
  
file_path: ReAgent/reagent/models/categorical_dqn.py
imports: import torch
  import torch.nn.functional
  from reagent.core import types
  from reagent.models.base import ModelBase
label_desc: Use image node_id 1 to create a new CategoricalDQN object from inherited base classes: ModelBase with example: obj = CategoricalDQN(distributional_network)