tensorflow tf.equal() broadcasting用法补充

tf.equal()

逐元素判断Tensor x,y是否相等，是则为True，否则为False

equal(x, y, name=None)

大部分博客都只是如下一个简单的例子：

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

实际上tf.equal()输入的x y不一定要维度一致，其支持broadcasting操作

如下例子：

import tensorflow as tf

a = [[0, 0, 3]]
b = [[0], [0], [3]]
with tf.Session() as sess:
    print(sess.run(tf.equal(a, b)))

结果(a的每列会与b的每行的列元素逐一对比)：

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