o oi @sBddlmZmZddlmZmZdededededef dd Zd S) )Tensorstack)KORNIA_CHECK_IS_COLORKORNIA_CHECK_IS_TENSORimager_shiftg_shiftb_shiftreturncCsTt|t|d|j|||g}|t|dddddd|jddd}|S)zShift 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 )dimr)minmax)rrshaperviewtoclamp_)rrrr shiftsshiftedrL/home/ubuntu/.local/lib/python3.10/site-packages/kornia/enhance/shift_rgb.py shift_rgbs  ,rN) kornia.corerrkornia.core.checkrrrrrrrs"