ray.rllib.models.torch.torch\_modelv2.TorchModelV2
==================================================

.. currentmodule:: ray.rllib.models.torch.torch_modelv2

.. autoclass:: TorchModelV2
   :show-inheritance:

   
   
   .. rubric:: Methods

   .. autosummary::
      :nosignatures:
      :toctree:

   
      ~TorchModelV2.__init__
      ~TorchModelV2.context
      ~TorchModelV2.custom_loss
      ~TorchModelV2.forward
      ~TorchModelV2.get_initial_state
      ~TorchModelV2.import_from_h5
      ~TorchModelV2.is_time_major
      ~TorchModelV2.last_output
      ~TorchModelV2.metrics
      
      ~TorchModelV2.value_function