code: def test_jit(self, device, dtype):
      angle = torch.tensor([90.0], device=device, dtype=dtype)
      batch_size, channels, height, width = 2, 3, 64, 64
      img = torch.ones(
          batch_size,
          channels,
          height,
          width,
          device=device,
          dtype=dtype,
      )
      rot = kornia.geometry.transform.Rotate(angle)
      rot_traced = torch.jit.trace(
          kornia.geometry.transform.Rotate(angle), img
      )
      assert_close(rot(img), rot_traced(img))
  
imports: import pytest
  import torch
  from torch.autograd import gradcheck
  import kornia
  import kornia.testing
  from kornia.testing import assert_close
label_desc: Use image node_id 5 for calling the TestRotate obj's underlying member method code with example usage: obj.test_jit(device, dtype) without return types