
    3j              
           S SK r S SKJrJr  S\ R                  S\ R                  S\ R                  S\ R                  S\ R                  4
S jrg)	    N)KORNIA_CHECK_IS_COLORKORNIA_CHECK_IS_TENSORimager_shiftg_shiftb_shiftreturnc                     [        U 5        [        U SU R                   35        XU/nU [        R                  " USS9R                  SSSS5      R                  U 5      -   R                  SSS9nU$ )z|Shift rgb channels.

Shift each image's channel by either r_shift for red, g_shift for green and b_shift for blue channels.
zwith shape    )dim   r   )minmax)r   r   shapetorchstackviewtoclamp_)r   r   r   r   shiftsshifteds         R/home/wildlama/miniconda3/lib/python3.13/site-packages/kornia/enhance/shift_rgb.py	shift_rgbr      ss    
 5!%;u{{m!<=(Fu{{6q166r1aCFFuMMUUZ[abUcGN    )r   kornia.core.checkr   r   Tensorr    r   r   <module>r      sO   &  KU\\ ELL 5<< Z_ZfZf kpkwkw r   