code: def one_hot_encode(array, num_of_classes):
      return np.eye(num_of_classes)[array.reshape(-1)]
  
imports: import gzip
  import numpy
  import struct
label_desc: Use image node_id 2 for calling a global function with example usage: one_hot_encode(array, num_of_classes) and returns: np