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