code: def forward(self, state: rlt.FeatureData):
      dist = self.log_dist(state).exp()
      q_values = (dist * self.support.to(dist.device)).sum(2)
      return q_values
  
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 3 for calling the CategoricalDQN obj's underlying member method code with example usage: obj.forward(state) and returns: q_values