tf.cond

版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。

本文链接： https://blog.csdn.net/qq_36825778/article/details/102768248

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 操作不执行