tf.cond()函数的用法

 

这个函数跟if...else...的功能很像,主要控制tensorflow中计算图的张量的流向。官网中有对函数参数的解释如下:

tf.cond(
    pred,
    true_fn=None,
    false_fn=None,
    strict=False,
    name=None,
    fn1=None,
    fn2=None
)
  • pred: A scalar determining whether to return the result of true_fn or false_fn.
  • true_fn: The callable to be performed if pred is true.
  • false_fn: The callable to be performed if pred is false.
  • strict: A boolean that enables/disables 'strict' mode; see above.
  • name: Optional name prefix for the returned tensors.

看的有点糊涂,用一个例子来解释下吧 !

import tensorflow as tf
x=tf.constant(2)
y=tf.constant(5)
flag=tf.constant(True)
op=tf.cond(flag,lambda :tf.add(x,y),lambda : tf.multiply(x,y))
with tf.Session() as sess:
    result=sess.run(op)
    print(result)

当flag为True时,执行‘加’操作,结果为7,当flag为Flase时,执行“乘”操作,结果为10

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值