class_docstr: Remove given dimensions of input of size 1.
  Operates on tensors.
  
  Parameters
  ----------
  *dims : List[int]
      Dimensions to squeeze.
code: def generate_code(self) -> Callable:
      def squeeze(inp, _):
          inp.squeeze_(*self.dims)
          return inp
  
      return squeeze
  
file_path: ffcv/ffcv/transforms/common.py
imports: from typing import Callable, Optional, Tuple
  from ..pipeline.allocation_query import AllocationQuery
  from ..pipeline.operation import Operation
  from ..pipeline.state import State
  from dataclasses import replace
label_desc: Use image node_id 2 for calling the Squeeze obj's underlying member method code with example usage: obj.generate_code() and returns: squeeze, inp