class_docstr: Test utils for Shunted Transformer
code: def test_mlp(self):
      dummy_input = torch.rand(1, 16, 3).to(self.device)
      model = Mlp(in_features=3, hidden_features=5, out_features=4).to(
          self.device
      )
      out = model(dummy_input, 4, 4)
      self.assertTrue(out.shape == (1, 16, 4))
  
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 3 for calling the TestUtils obj's underlying member method code with example usage: obj.test_mlp() without return types