code: def th_flatten(x):
      """Flatten tensor"""
      return x.contiguous().view(-1)
  
imports: import pickle
  import random
  from .optimizers import *
  from .callbacks import *
  import torch
  from pywick.callbacks import *
label_desc: Use image node_id 2 for calling a global function with example usage: th_flatten(x) and returns: x