import tensorflow as tf

      else:
        # Take body outputs for the last position only, and targets too.
        last_position_body_output = tf.expand_dims(
            body_output[:, -1, :, :], axis=[1])
