code: def test_squeeze_dims(self):
      @torch.jit.script
      def foo(x):
          return torch.ops.aten.squeeze(x, dim=0)
  
      input = next(foo.graph.inputs())
      input.setType(input.type().with_sizes([1, 5, 8]))
      torch._C._jit_pass_propagate_shapes_on_graph(foo.graph)
      self.assertEqual(
          next(foo.graph.outputs()).type().symbolic_sizes(), [5, 8]
      )
  
imports: import operator
  import unittest
  from textwrap import dedent
  import torch
  from torch import nn, Tensor
  from torch.testing import FileCheck
  from torch.testing._internal.common_methods_invocations import sample_inputs_cat_concat
  from torch.testing._internal.common_utils import make_tensor
  from torch.testing._internal.jit_utils import JitTestCase, execWrapper
  from typing import List, Any
label_desc: Use image node_id 30 for calling the TestSymbolicShapeAnalysis obj's underlying member method code with example usage: obj.test_squeeze_dims() and returns: torch