tf.equal函数

函数原型

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

函数说明

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

函数使用

>>> a = tf.constant([[1, 2], [3, 4]])
>>> a
<tf.Tensor: shape=(2, 2), dtype=int32, numpy=
array([[1, 2],
       [3, 4]])>
>>> b = 2
# 与单个数进行比较
>>> c = tf.equal(a, b)
>>> c
<tf.Tensor: shape=(2, 2), dtype=bool, numpy=
array([[False,  True],
       [False, False]])>
>>> b = tf.constant([1, 2])
>>> b
<tf.Tensor: shape=(2,), dtype=int32, numpy=array([1, 2])>
# 相当于与[[1, 2], [1, 2]]进行相同位置的元素比较
>>> c = tf.equal(a, b)
>>> c
<tf.Tensor: shape=(2, 2), dtype=bool, numpy=
array([[ True,  True],
       [False, False]])>
>>> b = tf.constant([[2, 2], [3, 3]])
# 相同位置的元素进行比较
>>> c = tf.equal(a, b)
>>> c
<tf.Tensor: shape=(2, 2), dtype=bool, numpy=
array([[False,  True],
       [ True, False]])>