tf.equal的使用

tf.equal(A, B)是对比这两个矩阵或者向量的相等的元素，如果是相等的那就返回True，反正返回False，返回的值的矩阵维度和A是一样的

[python] view plain copy

1. import tensorflow as tf  
2. import numpy as np  
3.   
4. A = [[1,3,4,5,6]]  
5. B = [[1,3,4,3,2]]  
6.   
7. with tf.Session() as sess:  
8.     print(sess.run(tf.equal(A, B)))  

输出：

[[ True  True  True False False]]