Tensorflow 如何判断两个tensor相等

Question

Tensorflow 如何判断两个tensor相等？
tf.equal()会判断两个tensor的每个元素是否相等，例如：
a=[1,0] b=[0,0] tf.equal(a,b) 结果为[False,True]

现在我想整体判断两个tensor是否相等，比如:
a=[1,0] b=[0,0] 判断结果为False
a=[1,0] b=[1,0] 判断结果为True
即两个tensor元素完全相同，则返回一个True，否则返回False，该如何做？

Answer 1

我不知道有没有专门的函数，不过你可以将 tf.equal(a,b)的结果由布尔值转换为浮点数，再取平均值
实现方法如下：

with tf.Session() as sess:
    sess.run(tf.reduce_mean(tf.cast(tf.equal(a, b),'float'))) 

如果结果为1，则表示a,b两个tensor相等，否则不相等
当然这只是个下策，可能会存在一些问题，因为我也不知道更好的方法