Tenserflow中的where和greater实现对两个矩阵的比较
发布日期:2021-05-07 04:21:56 浏览次数:32 分类:精选文章

本文共 441 字,大约阅读时间需要 1 分钟。

有时候会需要对矩阵进行大小的比较,或者是选择出矩阵中值较大的部分,此时就可以使用greater和where函数来完成。


第一步,先定义两个矩阵

v1 = tf.constant([[1.0,2.0],[13.0,14.0]])v2 = tf.constant([[5.0,6.0],[7.0,8.0]])

使用greater函数可以对矩阵进行点对点的比较,并且得出True(更大)或者False(更小)作为结果

我们将v1和v2进行比较,

tf.greater(v1, v2)

得到的结果是

[[False False]

 [ True  True]]

而where函数可以通过True和False,选择两个矩阵,这一个函数具有三个参数,其中后两者是输入的矩阵。通过搭配greater使用,即可选择出两个矩阵中最大或者最小的数值

print(sess.run(tf.where(tf.greater(v1,v2),v1,v2)))

运行后的结果

[[ 5.  6.]

 [13. 14.]]

上一篇:说说我为什么讨厌嵌入式相关开发,还有物联网
下一篇:Tensorflow架构中的matmul矩阵乘法和mul点乘法

发表评论

最新留言

第一次来,支持一个
[***.219.124.196]2025年04月08日 05时13分45秒