code: def test_get_shape(self) -> None:
      list_x = [torch.zeros(2, 4), torch.zeros(3, 3)]
      shapes = get_shape(list_x)
      expected_shapes = [
          torch.zeros(2, 4).size(),
          torch.zeros(3, 3).size(),
      ]
      for shape, expected in zip(shapes, expected_shapes):
          self.assertEqual(shape, expected)
  
      dict_x = {"x1": torch.zeros(2, 4), "x2": torch.zeros(3, 3)}
      shapes = get_shape(dict_x)
      expected_shapes = {
          "x1": torch.zeros(2, 4).size(),
          "x2": torch.zeros(3, 3).size(),
      }
      for k, shape in shapes.items():
          self.assertEqual(shape, expected_shapes[k])
  
      list_dict_x = [
          {"x1": torch.zeros(2, 4), "x2": torch.zeros(3, 3)},
          {"x1": torch.zeros(3, 4), "x2": torch.zeros(4, 5)},
      ]
      shapes = get_shape(list_dict_x)
      expected_shapes = [
          {
              "x1": torch.zeros(2, 4).size(),
              "x2": torch.zeros(3, 3).size(),
          },
          {
              "x1": torch.zeros(3, 4).size(),
              "x2": torch.zeros(4, 5).size(),
          },
      ]
      for shape, expected in zip(shapes, expected_shapes):
          for k, s in shape.items():
              self.assertEqual(s, expected[k])
  
imports: import unittest
  import torch
  import torch.nn
  from classy_vision.generic.profiler import compute_activations, compute_flops, count_params, get_shape
  from classy_vision.models import build_model
  from test.generic.config_utils import get_test_model_configs
label_desc: Use image node_id 1 for calling the TestHelperFunctions obj's underlying member method code with example usage: obj.test_get_shape() without return types