code: def test_autumn(device, dtype):
      cm = AUTUMN(num_colors=254, device=device, dtype=dtype)
      colors = cm.colors
  
      actual = colors[..., 0]
  
      expected = tensor([1, 0, 0], device=device, dtype=dtype)
  
      assert_close(actual, expected)
  
      actual = colors[..., 127]
      expected = tensor(
          [1, 0.5019997358322144, 0], device=device, dtype=dtype
      )
      assert_close(actual, expected)
  
      actual = colors[..., -1]
      expected = tensor([1, 1, 0], device=device, dtype=dtype)
      assert_close(actual, expected)
  
file_path: kornia/test/color/test_colormap.py
imports: import pytest
  import torch
  from torch.autograd import gradcheck
  from kornia.color import AUTUMN, ApplyColorMap, apply_colormap
  from kornia.core import tensor
  from kornia.testing import BaseTester, assert_close, tensor_to_gradcheck_var
label_desc: Use image node_id 1 for calling a global function with example usage: test_autumn(device, dtype) without return types