def Body(i, outputs): x = tf.gather(inputs, i) # pylint: disable=cell-var-from-loop outputs = outputs.write(i, x) return i + 1, outputs
def Body(i, outputs): x = tf.gather(inputs, i) outputs = outputs.write(i, x) return i + 1, outputs