tf.equal()函数介绍和示例

tf.equal()函数介绍和示例

tf.equal(A, B, name=None)

释义：对比 A，B 矩阵或向量的元素，若相等则返回True，否则返回False

注意：不同数据类型比较会报错，需要转换为相同数据类型进行比较

示例：

import tensorflow as tf

X = tf.constant([1.,2.,3.,4.,5.,6.], dtype=tf.float32)
Y = tf.constant([1.,2.,3.,7.,8.,9.], dtype=tf.float32)

with tf.Session() as sess:
    print(sess.run(tf.equal(X, Y)))

[ True  True  True False False False]