
    3j                    t    S SK Jr  S SKrS SKJr  S SKJrJrJrJr  SS	S jjr	 " S S\R                  5      rg)
    )annotationsN)nn)KORNIA_CHECKKORNIA_CHECK_IS_TENSORKORNIA_CHECK_SAME_DEVICEKORNIA_CHECK_SAME_SHAPEc                j   [        U 5        [        U5        [        X5        [        X5        [        US;   SU 35        [        R
                  " S[        R                  " X-
  5      -  5      nUS:X  a  UR                  5       nU$ US:X  a  UR                  5       nU$ US:X  d  Uc   U$ [        S5      e)a<  Criterion that computes the Cauchy [2] (aka. Lorentzian) loss.

According to [1], we compute the Cauchy loss as follows:

.. math::

    \text{WL}(x, y) = log(\frac{1}{2} (x - y)^{2} + 1)

Where:
   - :math:`x` is the prediction.
   - :math:`y` is the target to be regressed to.

Reference:
    [1] https://arxiv.org/pdf/1701.03077.pdf
    [2] https://files.is.tue.mpg.de/black/papers/cviu.63.1.1996.pdf

Args:
    img1: the predicted torch.Tensor with shape :math:`(*)`.
    img2: the target torch.Tensor with the same shape as img1.
    reduction: Specifies the reduction to apply to the
      output: ``'none'`` | ``'mean'`` | ``'sum'``. ``'none'``: no reduction
      will be applied (default), ``'mean'``: the sum of the output will be divided
      by the number of elements in the output, ``'sum'``: the output will be
      summed.

Return:
    a scalar with the computed loss.

Example:
    >>> img1 = torch.randn(2, 3, 32, 32, requires_grad=True)
    >>> img2 = torch.randn(2, 3, 32, 32)
    >>> output = cauchy_loss(img1, img2, reduction="mean")
    >>> output.backward()

)meansumnoneNz/Given type of reduction is not supported. Got: g      ?r
   r   r   zInvalid reduction option.)
r   r   r   r   torchlog1psquarer
   r   NotImplementedError)img1img2	reductionlosss       N/home/wildlama/miniconda3/lib/python3.13/site-packages/kornia/losses/cauchy.pycauchy_lossr      s    H 4 4 D'T(226efoep4q
 ;;sU\\$+667D Fyy{ K 
e	xxz K 
f		 1 K ""=>>    c                  >   ^  \ rS rSrSrSSU 4S jjjrSS jrSrU =r$ )	
CauchyLossZ   a;  Criterion that computes the Cauchy [2] (aka. Lorentzian) loss.

According to [1], we compute the Cauchy loss as follows:

.. math::

    \text{WL}(x, y) = log(\frac{1}{2} (x - y)^{2} + 1)

Where:
   - :math:`x` is the prediction.
   - :math:`y` is the target to be regressed to.

Reference:
    [1] https://arxiv.org/pdf/1701.03077.pdf
    [2] https://files.is.tue.mpg.de/black/papers/cviu.63.1.1996.pdf

Args:
    reduction: Specifies the reduction to apply to the
      output: ``'none'`` | ``'mean'`` | ``'sum'``. ``'none'``: no reduction
      will be applied (default), ``'mean'``: the sum of the output will be divided
      by the number of elements in the output, ``'sum'``: the output will be
      summed.

Shape:
    - img1: the predicted torch.Tensor with shape :math:`(*)`.
    - img2: the target torch.Tensor with the same shape as img1.

Example:
    >>> criterion = CauchyLoss(reduction="mean")
    >>> img1 = torch.randn(2, 3, 32, 2107, requires_grad=True)
    >>> img2 = torch.randn(2, 3, 32, 2107)
    >>> output = criterion(img1, img2)
    >>> output.backward()

c                .   > [         TU ]  5         Xl        g )N)super__init__r   )selfr   	__class__s     r   r   CauchyLoss.__init__   s    "r   c                *    [        XU R                  S9$ )N)r   r   r   )r   r   )r   r   r   s      r   forwardCauchyLoss.forward   s    4>>JJr   )r   r   )r   strreturnNone)r   torch.Tensorr   r(   r&   r(   )	__name__
__module____qualname____firstlineno____doc__r   r"   __static_attributes____classcell__)r   s   @r   r   r   Z   s    "H# #K Kr   r   r$   )r   r(   r   r(   r   r%   r&   r(   )
__future__r   r   r   kornia.core.checkr   r   r   r   r   Moduler    r   r   <module>r4      s.   $ #   u u=@*K *Kr   