Tensorflow——tf.cond()的用法

定义格式

tf.cond(pred, fn1, fn2, name=None)
Return :either fn1() or fn2() based on the boolean predicate pred.(注意这里，也就是说'fnq'和‘fn2’是两个函数)

在TensorFlow中，tf.cond()类似于c语言中的if...else...，用来控制数据流向

例子

import tensorflow as tf
a=tf.constant(2)    
b=tf.constant(3)    
x=tf.constant(4)    
y=tf.constant(5)    
z = tf.multiply(a, b)    
result = tf.cond(x < y, lambda: tf.add(x, z), lambda: tf.square(y))    
with tf.Session() as session:    
    print(result.eval())

输出
10