return tf.nn.relu(x + c)

def condition(x):
    return tf.reduce_sum(x) < 100
x = tf.Variable(tf.constant(0, shape=[2, 2]))
with tf.Session():
    tf.initialize_all_variables().run()
result = tf.while_loop(condition, body, [x])
print(result.eval())
Как использовать tf.while_loop() в тензорном потоке