code: def find_tensor_by_shape(
      target_shape: Tuple, only_param: bool = True
  ) -> bool:
      """Find a tensor from the heap
  
      Args:
          target_shape (tuple):
              Tensor shape to locate.
          only_param (bool):
              Only match Parameter type (e.g. for weights).
  
      Returns:
          (bool):
              Return True if found.
      """
      for obj in gc.get_objects():
          try:
              # Only need to check parameter type objects if asked.
              if (
                  only_param
                  and "torch.nn.parameter.Parameter"
                  not in str(type(obj))
              ):
                  continue
              if torch.is_tensor(obj) or (
                  hasattr(obj, "data") and torch.is_tensor(obj.data)
              ):
                  if obj.shape == target_shape:
                      return True
          except Exception as e:
              pass
      return False
  
label_desc: Use image node_id 1 for calling a global function with example usage: find_tensor_by_shape(target_shape, only_param) and returns: False, True