code: def th_bc_flatten(x):
      """
      Flatten tensor, leaving batch and channel dims intact.
      Assumes BCHW format
      """
      return x.contiguous().view(x.size(0), x.size(1), -1)
  
imports: import pickle
  import random
  from .optimizers import *
  from .callbacks import *
  import torch
  from pywick.callbacks import *
label_desc: Use image node_id 4 for calling a global function with example usage: th_bc_flatten(x) and returns: x