code: def NCHW2NHWC(tensor):
      assert tensor.ndim >= 2
      return tensor.transpose(
          (0,) + tuple(range(2, tensor.ndim)) + (1,)
      )
  
imports: from caffe2.proto import caffe2_pb2
  from google.protobuf.message import DecodeError, Message
  from google.protobuf import text_format
  import sys
  import collections
  import copy
  import functools
  import numpy
label_desc: Use image node_id 18 for calling a global function with example usage: NCHW2NHWC(tensor) and returns: tensor