tf.cond和tf.case

tensorflow的逻辑控制
关于tf.case理解的不是很好:
https://stackoverflow.com/questions/41910073/tensorflow-tf-case-f-default-should-do-nothing
待明确后再补充
关于tf.cond,这里给出几个示例,对我来说,例子就是快速明白用法的途径

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())

运行的结果是25
根据第7行代码,如果x>y为true,就执行tf.add(x, z), 否则就执行tf.square(y)
示例二:
这里写图片描述

运行的结果是:2
如果注释第8行,取消注释第9行,此时运行的结果是1

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值
>