code: def test_shape(self, device):
      inp = torch.ones(1, 1, 32, 32, device=device)
      tfeat = TFeat().to(device)
      tfeat.eval()  # batchnorm with size 1 is not allowed in train mode
      out = tfeat(inp)
      assert out.shape == (1, 128)
  
file_path: kornia/test/feature/test_tfeat.py
imports: import pytest
  import torch
  from torch.autograd import gradcheck
  import kornia.testing
  from kornia.feature import TFeat
  from kornia.testing import assert_close
label_desc: Use image node_id 1 for calling the TestTFeat obj's underlying member method code with example usage: obj.test_shape(device) without return types