docstr: test the grid_partition
  Returns:None
code: def test_window_partitionn(self):
      """
      test the grid_partition
      Returns:None
      """
      x = torch.randn(1, 3, 28, 28)
      out = window_partition(x)
      self.assertEqual(out.shape, (16, 7, 7, 3))
  
file_path: towhee/tests/unittests/models/max_vit/test_max_vit_utils.py
imports: import unittest
  import torch
  from towhee.models.max_vit.max_vit_utils import grid_partition, window_partition, get_relative_position_index, window_reverse, grid_reverse
label_desc: Use image node_id 2 for calling the MaxVitUtilsTest obj's underlying member method code with example usage: obj.test_window_partitionn() without return types