code: def shift_right(x):
      """Shift the input over by one and a zero to the front.
  
      Args:
        x: The [mb, time, channels] tensor input.
  
      Returns:
        x_sliced: The [mb, time, channels] tensor output.
      """
      shape = x.get_shape().as_list()
      x_padded = tf.pad(x, [[0, 0], [1, 0], [0, 0]])
      x_sliced = tf.slice(
          x_padded, [0, 0, 0], tf.stack([-1, shape[1], -1])
      )
      x_sliced.set_shape(shape)
      return x_sliced
  
file_path: magenta/magenta/models/nsynth/wavenet/masked.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import tensorflow.compat.v1
label_desc: Use image node_id 1 for calling a global function with example usage: shift_right(x) and returns: x_sliced