code: def flatten_tensor(x: torch.Tensor):
      original_shape = x.shape
      x = x.view([-1])
      return x, original_shape
  
file_path: espnet/espnet2/asr/transducer/rnnt_multi_blank/utils/rnnt_helper.py
imports: import math
  from typing import Optional, Tuple
  import torch
  from numba import cuda
  from espnet2.asr.transducer.rnnt_multi_blank.utils import global_constants
label_desc: Use image node_id 12 for calling a global function with example usage: flatten_tensor(x) and returns: x,  original_shape