
    3j                    :    S SK Jr  S SKJr  S SKr      SS jrg)    )annotations)OptionalNc                    Uc  U S 4$ X:g  nUR                  5       (       a  U S 4$ U R                  X R                  S5      5      n X4$ )N   )allwhere	new_zeros)targetignore_indextarget_masks      N/home/wildlama/miniconda3/lib/python3.13/site-packages/kornia/losses/_utils.pymask_ignore_pixelsr      sW     t|(Kt| \\+'7'7':;F    )r
   ztorch.Tensorr   zOptional[int]returnz+tuple[torch.Tensor, Optional[torch.Tensor]])
__future__r   typingr   torchr    r   r   <module>r      s-   $ #  (50r   