class_docstr: Test utils for Shunted Transformer
code: def test_head_patch(self):
      fake_img = torch.rand(1, 3, 4, 4).to(self.device)
      model = HeadPatchEmbed(1).to(self.device)
      out, h, w = model(fake_img)
      self.assertEqual(h, w, 1)
      self.assertTrue(out.shape == (1, 1, 64))
  
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 2 for calling the TestUtils obj's underlying member method code with example usage: obj.test_head_patch() without return types