tf.where()用法

最新推荐文章于 2022-10-12 14:39:55 发布

Eric_LH

最新推荐文章于 2022-10-12 14:39:55 发布

阅读量1.2k

点赞数

分类专栏： tensorflow实战

本文链接：https://blog.csdn.net/Eric_LH/article/details/83794225

版权

tensorflow实战专栏收录该内容

42 篇文章 0 订阅

订阅专栏

import tensorflow as tf 
x = [[1,2,3],[4,5,6]] 
y = [[7,8,9],[10,11,12]] 
condition3 = [[True,False,False], [False,True,True]]
condition4 = [[True,False,False], [True,True,False]] 
with tf.Session() as sess: 
    print(sess.run(tf.where(condition3,x,y)))
    print(sess.run(tf.where(condition4,x,y)))

output:
1， [[ 1 8 9]
[10 5 6]]
2， [[ 1 8 9]
[ 4 5 12]]

condition， x, y 相同维度，condition是bool型值，True/False

返回值是对应元素，condition中元素为True的元素替换为x中的元素，为False的元素替换为y中对应元素

x只负责对应替换True的元素，y只负责对应替换False的元素，x，y各有分工

由于是替换，返回值的维度，和condition，x ， y都是相同的。