code: def test_shape(self, device):
      inp = torch.ones(1, 1, 3, 4, 4, device=device)
      nms = kornia.geometry.subpix.NonMaximaSuppression3d((3, 3, 3)).to(
          device
      )
      assert nms(inp).shape == inp.shape
  
file_path: kornia/test/geometry/subpix/test_nms.py
imports: import torch
  from torch.autograd import gradcheck
  import kornia
  import kornia.testing
  from kornia.testing import assert_close
label_desc: Use image node_id 1 for calling the TestNMS3d obj's underlying member method code with example usage: obj.test_shape(device) without return types