class_docstr: Action distribution that operates on multiple, possibly nested actions.
class_docstr_tok: ["Action","distribution","that","operates","on","multiple",",","possibly","nested","actions","."]
code: def sample(self):
      child_distributions_struct = tree.unflatten_as(
          self._original_struct, self._flat_child_distributions
      )
      return tree.map_structure(
          lambda s: s.sample(), child_distributions_struct
      )
  
imports: import gymnasium
  import numpy
  from typing import Optional, List, Mapping, Iterable, Dict
  import tree
  import abc
  from ray.rllib.models.distributions import Distribution
  from ray.rllib.utils.annotations import override, DeveloperAPI
  from ray.rllib.utils.framework import try_import_torch
  from ray.rllib.utils.typing import TensorType, Union, Tuple
label_desc: Use image node_id 6 for calling the TorchMultiDistribution obj's underlying member method code with example usage: obj.sample() and returns: tree