tf.where(a,b,c)函数

最新推荐文章于 2023-03-24 19:06:59 发布

放下扳手&拿起键盘

最新推荐文章于 2023-03-24 19:06:59 发布

阅读量2.5k

点赞数 2

CC 4.0 BY-SA版权

分类专栏： tensorflow

本文链接：https://blog.youkuaiyun.com/william_hehe/article/details/78636624

tensorflow 专栏收录该内容

43 篇文章

订阅专栏

本文介绍了 TensorFlow 中 tf.where(a,b,c) 函数的功能与用法。该函数根据条件 a 的真假选择 b 或 c 输出。通过实例展示了如何使用该函数进行条件判断，并返回相应数值。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

tf.where(a,b,c)函数：
功能：当a输出结果为true时，tf.where(a,b,c)函数会选择b值输出。
当a输出结果为false时，tf.where(a,b,c)函数会选择c值输出。
例子：

import tensorflow as tf
v1=tf.constant([1.0,2.0,3.0,4.0])
v2=tf.constant([4.0,3.0,2.0,1.0])

with tf.Session() as sess:
    init=tf.global_variables_initializer()
    sess.run(init)
    print(sess.run(tf.greater(v1,v2)))
    print(sess.run(tf.where(tf.greater(v1,v2),v1,v2)))