tf.cond

免責事項:この記事はブロガーオリジナル記事です、続くBY-SAのCC 4.0を著作権契約、複製、元のソースのリンクと、この文を添付してください。
このリンク: 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を行いません

おすすめ

転載: blog.csdn.net/qq_36825778/article/details/102768248