tf.equal()函数用法及解析

tf.equal()函数学习

tf.equal()函数
函数定义

equal(x, y, name=None)

函数作用及参数解读
equal，相等的意思。顾名思义，该函数用来判断相等
tf.equal(A, B)是对比这两个矩阵或者向量的相等的元素，如果是相等的那就返回True，反正返回False，返回的值的矩阵维度和A是一样的
由于是逐元素的对比，所以x和y的维度也要相同
其中：

• x,y:要比较的两个张量
• name:操作名称，默认null

函数实例

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]]

可以看到，输出的矩阵维度不变

注意
equal()函数是逐元素比较，其返回结果是一个逐元素的比较结果（Ture/False）矩阵，如果需要其他形式的输出，记得转换。

参考文章
tenflow 入门 tf.equal(）用法