tf.equal函数

tf.math.equal(
    x, y, name=None
)

判断两个张量在相对应的元素上是否相同，结果用bool值来表示。

>>> a = tf.constant([[1, 2], [3, 4]])
>>> a

>>> b = 2
# 与单个数进行比较
>>> c = tf.equal(a, b)
>>> c

>>> b = tf.constant([1, 2])
>>> b

# 相当于与[[1, 2], [1, 2]]进行相同位置的元素比较
>>> c = tf.equal(a, b)
>>> c

>>> b = tf.constant([[2, 2], [3, 3]])
# 相同位置的元素进行比较
>>> c = tf.equal(a, b)
>>> c