pyepo.func.abcmodule
====================

.. py:module:: pyepo.func.abcmodule

.. autoapi-nested-parse::

   Abstract autograd optimization module



Classes
-------

.. autoapisummary::

   pyepo.func.abcmodule.optModule


Module Contents
---------------

.. py:class:: optModule(optmodel, processes=1, solve_ratio=1, reduction='mean', dataset=None, require_solpool=False)

   Bases: :py:obj:`torch.nn.Module`


   An abstract module for the learning to rank losses, which measure the difference in how the predicted cost
   vector and the true cost vector rank a pool of feasible solutions.


   .. py:attribute:: optmodel


   .. py:attribute:: processes


   .. py:attribute:: solve_ratio
      :value: 1



   .. py:attribute:: solpool
      :value: None



   .. py:attribute:: _solset


   .. py:attribute:: reduction
      :value: 'mean'



   .. py:method:: forward(pred_cost, true_cost)
      :abstractmethod:


      Forward pass



   .. py:method:: _reduce(loss)

      Apply reduction to loss tensor