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

    原文作者:SpareNoEfforts
    原文地址: https://www.jianshu.com/p/8b35a0a7025d
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞