TensorFlow if语句 tensor 和 非tensor 比较

import tensorflow as tf

condition = tf.placeholder(tf.int32, name="condition")

A = tf.constant(value=123)
B = tf.constant(value=321)

def func1():
    return A

def func2():
    return B

y = tf.cond(condition > 0, func1, func2) # tensor 和 非tensor 比较
 
sess = tf.Session()

feed_dict = {condition:1}
print(sess.run(y,feed_dict=feed_dict))

feed_dict = {condition:-1}
print(sess.run(y,feed_dict=feed_dict))

打印结果:
123
321

©️2020 CSDN 皮肤主题: 创作都市 设计师: CSDN官方博客 返回首页
实付0元
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值