class_docstr: Test utils for Shunted Transformer
code: def test_block(self):
      dummy_input = torch.rand(1, 4, 6).to(self.device)
  
      model1 = Block(dim=6, num_heads=3).to(self.device)
      outs1 = model1(dummy_input, 2, 2)
      self.assertTrue(outs1.shape == (1, 4, 6))
  
      model2 = Block(dim=6, num_heads=3, drop_path=0.2).to(self.device)
      outs2 = model2(dummy_input, 2, 2)
      self.assertTrue(outs2.shape == (1, 4, 6))
  
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 5 for calling the TestUtils obj's underlying member method code with example usage: obj.test_block() without return types