code: def test_shape(self, device, batch_size, dtype):
      inp = torch.ones(
          batch_size, 1, 64, 64, device=device, dtype=dtype
      )
      sold2 = SOLD2_detector(pretrained=False).to(device, dtype)
      out = sold2(inp)
      assert out["junction_heatmap"].shape == (batch_size, 64, 64)
      assert out["line_heatmap"].shape == (batch_size, 64, 64)
  
file_path: kornia/test/feature/test_sold2.py
imports: import pytest
  import torch
  from torch.autograd import gradcheck
  import kornia.testing
  from kornia.feature.sold2 import SOLD2, SOLD2_detector
  from kornia.testing import assert_close
label_desc: Use image node_id 1 for calling the TestSOLD2_detector obj's underlying member method code with example usage: obj.test_shape(device, batch_size, dtype) without return types