tf.cond

tf.cond函数的使用

函数描述：
如果pred 为 true 则返回 true_fn() ,否则返回 false_fn()

cond ( 
    pred , 
    true_fn = None , 
    false_fn = None , 
    strict = False , 
    name = None , 
    fn1 = None , 
    fn2 = None
 )

示例：

z = tf.multiply( a , b ) 
result = tf.cond( X < Y, lambda：tf.add( X ,Z), lambda： tf.square( Y ))

如果X < Y，tf.add 将执行，并且 tf.square 操作不执行