定义格式
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