foscat.loss_backend_torch
=========================

.. py:module:: foscat.loss_backend_torch


Classes
-------

.. autoapisummary::

   foscat.loss_backend_torch.loss_backend


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

.. py:class:: loss_backend(backend, curr_gpu, mpi_rank)

   .. py:attribute:: bk


   .. py:attribute:: curr_gpu


   .. py:attribute:: mpi_rank


   .. py:method:: check_dense(data, datasz)


   .. py:method:: loss(x, batch, loss_function, KEEP_TRACK)