a d^@s<ddlZddlmZmZejejejejejdddZdS)N)KORNIA_CHECK_IS_COLORKORNIA_CHECK_IS_TENSOR)imager_shiftg_shiftb_shiftreturncCsRt|t|d|j|||g}|t|dddd|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 r)minmax)rrshapetorchTensorviewtoZclamp_)rrrrZshiftsZshiftedra/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/kornia/enhance/shift_rgb.py shift_rgbs  *r)rZkornia.testingrrrrrrrrs