code: def test_rgb_shift(self, device, dtype):
      r_shift, g_shift, b_shift = (
          torch.Tensor([0.1]),
          torch.Tensor([0.3]),
          torch.Tensor([-0.3]),
      )
      image = torch.tensor(
          [
              [[[0.2, 0.0]], [[0.3, 0.5]], [[0.4, 0.7]]],
              [[[0.2, 0.7]], [[0.0, 0.8]], [[0.2, 0.3]]],
          ],
          device=device,
          dtype=dtype,
      )
      shifted = kornia.enhance.shift_rgb(
          image, r_shift, g_shift, b_shift
      )
      expected = torch.tensor(
          [
              [[[0.3, 0.1]], [[0.6, 0.8]], [[0.1, 0.4]]],
              [[[0.3, 0.8]], [[0.3, 1.0]], [[0.0, 0.0]]],
          ],
          device=device,
          dtype=dtype,
      )
  
      utils.assert_close(shifted, expected)
  
file_path: kornia/test/enhance/test_shift_rgb.py
imports: import pytest
  import torch
  from torch.autograd import gradcheck
  import kornia
  import kornia.testing
label_desc: Use image node_id 6 for calling the TestRGBShift obj's underlying member method code with example usage: obj.test_rgb_shift(device, dtype) without return types