code: def wrap(image):
      """Returns 'image' with an extra channel set to all 1s."""
      shape = tf.shape(image)
      extended_channel = tf.ones([shape[0], shape[1], 1], image.dtype)
      extended = tf.concat([image, extended_channel], 2)
      return extended
  
imports: import math
  import tensorflow
  import tensorflow_addons
label_desc: Use image node_id 2 for calling a global function with example usage: wrap(image) and returns: extended