o oi@s\ddlmZddlmZddlmZddlmZmZm Z m Z ddd d Z GdddeZ dS)) annotations)Tensor)Module) KORNIA_CHECKKORNIA_CHECK_IS_TENSORKORNIA_CHECK_SAME_DEVICEKORNIA_CHECK_SAME_SHAPEnoneimg1rimg2 reductionstrreturncCst|t|t||t||t|dvd|dd||d}|dkr2|}|S|dkr<|}|S|dkrC |Std ) aCriterion that computes the Welsch [2] (aka. Leclerc [3]) loss. According to [1], we compute the Welsch loss as follows: .. math:: \text{WL}(x, y) = 1 - exp(-\frac{1}{2} (x - y)^{2}) 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://www.tandfonline.com/doi/abs/10.1080/03610917808812083 [3] https://link.springer.com/article/10.1007/BF00054839 Args: img1: the predicted tensor with shape :math:`(*)`. img2: the target 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 = welsch_loss(img1, img2, reduction="mean") >>> output.backward() )meansumr z/Given type of reduction is not supported. Got: g?grrr zInvalid reduction option.)rrrrexprrNotImplementedError)r r r lossrH/home/ubuntu/.local/lib/python3.10/site-packages/kornia/losses/welsch.py welsch_losss %  rcs.eZdZdZddfdd Zdd d ZZS) WelschLossaCriterion that computes the Welsch [2] (aka. Leclerc [3]) loss. According to [1], we compute the Welsch loss as follows: .. math:: \text{WL}(x, y) = 1 - exp(-\frac{1}{2} (x - y)^{2}) 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://www.tandfonline.com/doi/abs/10.1080/03610917808812083 [3] https://link.springer.com/article/10.1007/BF00054839 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 tensor with shape :math:`(*)`. - img2: the target tensor with the same shape as img1. Example: >>> criterion = WelschLoss(reduction="mean") >>> img1 = torch.randn(2, 3, 32, 1904, requires_grad=True) >>> img2 = torch.randn(2, 3, 32, 1904) >>> output = criterion(img1, img2) >>> output.backward() r r r rNonecst||_dS)N)super__init__r )selfr  __class__rrrs  zWelschLoss.__init__r rr cCst|||jdS)N)r r r )rr )rr r rrrforwardszWelschLoss.forwardr )r r rr)r rr rrr)__name__ __module__ __qualname____doc__rr __classcell__rrrrrYs%rNr )r rr rr r rr) __future__rtorchr kornia.corerkornia.core.checkrrrrrrrrrrs    ?