【TensorFlow】【API】tf.math.equal() 函数讲解

1.作用

⭐️ 对相同shape的两个张量A和B，张量A和张量B对应位置的元素一一对比，判断对应位置的两个元素是否相等，相等，则在新tensor N中相应位置为True，否则为False。 其中，张量A，B，N的shape是一样的。
⭐️ 对一个张量A和一个常数C做对比，对张量A中每个元素都与常数C进行对比，判断这些位置的元素是否与常数C相等，相等，则在新的tensor N中相应位置为True，否则为False。其中，张量A和N的shape是一样的。

2.参数详解

⭐️ API 形式如下

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

⭐️ 参数说明如下表：

参数名	说明
x	张量A，也可以为常数
y	张量B，也可以为常数
name	一般设置为None

3.举例

⭐️ 张量A 和 张量B，代码如下

a=[1, 2, 0, 0, 1]
b = [1, 0, 2, 3, 1]
n=tf.math.equal(a,b)
print(n)
# 输出结果为
tf.Tensor([ True False False False  True], shape=(5,), dtype=bool)

⭐️ 张量A 和 常量C，代码如下

a=[1, 2, 0, 0, 1]
c = 0
n=tf.math.equal(a,c)
print(n)
# 输出结果为
tf.Tensor([False False  True  True False], shape=(5,), dtype=bool)

4.总结

⭐️ 由上可知，当函数输入参数为，张量A和张量B时，我们可以看到，相同位置相等的，在新张量N中的位置为True， 不相等的为False，且张量A、张量B和张量N的shape一致

⭐️ 由上可知，当函数输入参数为，张量A和常量C时，我们可以看到，当和常量C相等的，在新张量N中的位置为True，不相等的为False，且张量A，和张量N的shape一致

5.参考资料

📗 1. 官方文档：tf.math.equal