code: def test_shape_batch(self, device, dtype):
      inp = torch.ones(16, 1, 16, 16, device=device, dtype=dtype)
      keynet = KeyNet().to(device, dtype)
      out = keynet(inp)
      assert out.shape == inp.shape
  
file_path: kornia/test/feature/test_keynet.py
imports: import torch
  from torch.autograd import gradcheck
  import kornia.testing
  from kornia.feature import KeyNet
label_desc: Use image node_id 2 for calling the TestKeyNet obj's underlying member method code with example usage: obj.test_shape_batch(device, dtype) without return types