class_docstr: Test utils for Shunted Transformer
code: def test_overlap_patch(self):
      test_shape = (1, 3, 4, 4)
      fake_img = torch.rand(test_shape).to(self.device)
      model = OverlapPatchEmbed(
          img_size=test_shape[-1], patch_size=2, stride=1, embed_dim=5
      ).to(self.device)
      out, h, w = model(fake_img)
      self.assertEqual(h, w, 5)
      self.assertTrue(out.shape == (1, 25, 5))
  
file_path: towhee/tests/unittests/models/shunted_transformer/test_utils.py
imports: import unittest
  import torch
  from towhee.models.shunted_transformer import OverlapPatchEmbed, Mlp, Attention, Block, HeadPatchEmbed
label_desc: Use image node_id 1 for calling the TestUtils obj's underlying member method code with example usage: obj.test_overlap_patch() without return types