code: def test_batch2_n2(self, device, dtype):
      sample = torch.zeros(2, 2, 2, 3, device=device, dtype=dtype)
      sample[0, 0, 0, 0] = 1e16  # top-left
      sample[0, 1, 0, -1] = 1e16  # top-right
      sample[1, 0, -1, 0] = 1e16  # bottom-left
      sample[1, 1, -1, -1] = 1e16  # bottom-right
  
      coord = kornia.geometry.subpix.spatial_soft_argmax2d(sample)
      assert_close(
          coord[0, 0, 0].item(), -1.0, atol=1e-4, rtol=1e-4
      )  # top-left
      assert_close(coord[0, 0, 1].item(), -1.0, atol=1e-4, rtol=1e-4)
      assert_close(
          coord[0, 1, 0].item(), 1.0, atol=1e-4, rtol=1e-4
      )  # top-right
      assert_close(coord[0, 1, 1].item(), -1.0, atol=1e-4, rtol=1e-4)
      assert_close(
          coord[1, 0, 0].item(), -1.0, atol=1e-4, rtol=1e-4
      )  # bottom-left
      assert_close(coord[1, 0, 1].item(), 1.0, atol=1e-4, rtol=1e-4)
      assert_close(
          coord[1, 1, 0].item(), 1.0, atol=1e-4, rtol=1e-4
      )  # bottom-right
      assert_close(coord[1, 1, 1].item(), 1.0, atol=1e-4, rtol=1e-4)
  
imports: import torch
  from torch.autograd import gradcheck
  from torch.nn.functional import mse_loss
  import kornia
  import kornia.testing
  from kornia.geometry.subpix.spatial_soft_argmax import _get_center_kernel2d, _get_center_kernel3d
  from kornia.testing import assert_close
label_desc: Use image node_id 7 for calling the TestSpatialSoftArgmax2d obj's underlying member method code with example usage: obj.test_batch2_n2(device, dtype) without return types