code: def bhwc2bchw(x):
      if isinstance(x, np.ndarray):
          pass
      else:
          raise
  
      if x.ndim == 3:
          return np.moveaxis(x, 2, 0)
      if x.ndim == 4:
          return np.moveaxis(x, 3, 1)
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from __future__ import unicode_literals
  import os
  import sys
  import pathlib
  from torch.utils.data.dataset import Subset
  import numpy
  import torch
  import torchvision.transforms
  import torchvision.datasets
  from advertorch.test_utils import LeNet5
label_desc: Use image node_id 12 for calling a global function with example usage: bhwc2bchw(x) and returns: np, np