tenflow 入门 tf.equal(）用法

转载于 2019-04-22 10:17:29 发布 · 288 阅读

文章标签：

本文详细介绍了TensorFlow中tf.equal()函数的使用方法，该函数用于比较两个张量的对应元素是否相等，返回一个布尔类型的张量。通过一个具体示例展示了如何在会话中运行tf.equal()来比较两个二维张量。

原文：https://blog.youkuaiyun.com/ustbbsy/article/details/79564529

tensorflow 中tf.equal()用法：

equal(x, y, name=None)
equal，相等的意思。顾名思义，就是判断，x, y 是不是相等，它的判断方法不是整体判断，

而是逐个元素进行判断，如果相等就是True，不相等，就是False。

由于是逐个元素判断，所以x，y 的维度要一致。

看个例子：

import tensorflow as tf
a = [[1,2,3],[4,5,6]]
b = [[1,0,3],[1,5,1]]
with tf.Session() as sess:
    print(sess.run(tf.equal(a,b)))

结果：

[[ True False  True]
 [ False  True False]]