code: def move_axis(x, source, dest):
      """Move axis from source to dest."""
      return tf.convert_to_tensor(tnp.moveaxis(x, source, dest))
  
imports: import functools
  import numpy
  import six
  import tensorflow.compat.v2
  from tensorflow.python.util import nest
label_desc: Use image node_id 20 for calling a global function with example usage: move_axis(x, source, dest) and returns: tf