class_docstr: The policy action distribution of an agent.
  
  Attributes:
      inputs: input vector to compute samples from.
      model (ModelV2): reference to model producing the inputs.
class_docstr_tok: ["The","policy","action","distribution","of","an","agent",".","Attributes",":","inputs",":","input","vector","to","compute","samples","from",".","model","(","ModelV2",")",":","reference","to","model","producing","the","inputs","."]
docstr: Initializes an ActionDist object.
  
  Args:
      inputs: input vector to compute samples from.
      model (ModelV2): reference to model producing the inputs. This
          is mainly useful if you want to use model variables to compute
          action outputs (i.e., for auto-regressive action distributions,
          see examples/autoregressive_action_dist.py).
docstr_tok: ["Initializes","an","ActionDist","object",".","Args",":","inputs",":","input","vector","to","compute","samples","from",".","model","(","ModelV2",")",":","reference","to","model","producing","the","inputs",".","This","is","mainly","useful","if","you","want","to","use","model","variables","to","compute","action","outputs","(","i.e.",",","for","auto-regressive","action","distributions",",","see","examples\/autoregressive_action_dist.py",")","."]
code: def __init__(self, inputs: List[TensorType], model: ModelV2):
      """Initializes an ActionDist object.
  
      Args:
          inputs: input vector to compute samples from.
          model (ModelV2): reference to model producing the inputs. This
              is mainly useful if you want to use model variables to compute
              action outputs (i.e., for auto-regressive action distributions,
              see examples/autoregressive_action_dist.py).
      """
      self.inputs = inputs
      self.model = model
  
file_path: ray/python/ray/rllib/models/action_dist.py
imports: import numpy
  import gymnasium
  from ray.rllib.models.modelv2 import ModelV2
  from ray.rllib.utils.annotations import DeveloperAPI
  from ray.rllib.utils.typing import TensorType, List, Union, ModelConfigDict
label_desc: Use image node_id 1 to create a new ActionDistribution object  with example: obj = ActionDistribution(inputs, model)