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: Returns the required shape of an input parameter tensor for a
  particular action space and an optional dict of distribution-specific
  options.
  
  Args:
      action_space (gym.Space): The action space this distribution will
          be used for, whose shape attributes will be used to determine
          the required shape of the input parameter tensor.
      model_config: Model's config dict (as defined in catalog.py)
  
  Returns:
      model_output_shape (int or np.ndarray of ints): size of the
          required input vector (minus leading batch dimension).
docstr_tok: ["Returns","the","required","shape","of","an","input","parameter","tensor","for","a","particular","action","space","and","an","optional","dict","of","distribution-specific","options",".","Args",":","action_space","(","gym.Space",")",":","The","action","space","this","distribution","will","be","used","for",",","whose","shape","attributes","will","be","used","to","determine","the","required","shape","of","the","input","parameter","tensor",".","model_config",":","Model","'s","config","dict","(","as","defined","in","catalog.py",")","Returns",":","model_output_shape","(","int","or","np.ndarray","of","ints",")",":","size","of","the","required","input","vector","(","minus","leading","batch","dimension",")","."]
code: def required_model_output_shape(
      action_space: gym.Space, model_config: ModelConfigDict
  ) -> Union[int, np.ndarray]:
      """Returns the required shape of an input parameter tensor for a
      particular action space and an optional dict of distribution-specific
      options.
  
      Args:
          action_space (gym.Space): The action space this distribution will
              be used for, whose shape attributes will be used to determine
              the required shape of the input parameter tensor.
          model_config: Model's config dict (as defined in catalog.py)
  
      Returns:
          model_output_shape (int or np.ndarray of ints): size of the
              required input vector (minus leading batch dimension).
      """
      raise NotImplementedError
  
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 10 for calling the ActionDistribution obj's underlying member method code with example usage: obj.required_model_output_shape(action_space, model_config) without return types