Tensorflow(4)-control_flow_ops

該函數用來控制程序執行流，相當於if-else了

import tensorflow as tf
from tensorflow.python.ops import control_flow_ops

a = tf.constant(1)
b = tf.constant(3)

condition = tf.convert_to_tensor(False, dtype='bool')

rtval = control_flow_ops.cond(condition, 
                             lambda: a,
                             lambda: b)

with tf.Session() as sess:
    sess.run(tf.global_variables_initializer())
    print sess.run(rtval)

輸出爲 3