ggt.losses.aleatoric_loss
=========================

.. py:module:: ggt.losses.aleatoric_loss


Functions
---------

.. autoapisummary::

   ggt.losses.aleatoric_loss.aleatoric_loss


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

.. py:function:: aleatoric_loss(outputs, targets, average=True)

   Computes the aleatoric loss.
   Args:
       outputs: (tensor) - predicted outputs from the model
       targets: (tensor) - ground truth labels
       size_average: (bool) - if True, the losses are
              averaged over all elements of the batch
   Returns:
       aleatoric_loss: (tensor) - aleatoric loss