寻找两个张量的交集

Question

我在语义分割神经网络中用过dice系数是通过2xintersection/union计算的。 y_true 和 y_pred 的交集是由 tf.math.reduce_sum(y_pred*y_true) 找到的。请哪位高人帮我看看，两个张量的乘积怎么等于交集？

Answer 1

这是乘法变成交集的特例。

考虑一下，

y_true = [
[0,1,0],
[1,1,0],
[1,0,1]
]

和

y_pred = [
[0,1,1],
[1,0,0],
[1,0,1]
]

那么y_true * y_pred会是，

res = [
[0,1,0],
[1,0,0],
[1,0,1]

下一个tf.reduce_sum()给出res中所有1的和（即交集）。换句话说，只有当 y_true 和 y_pred 的那个位置都为 1 时，res 才会将一个元素设置为 1。

intersection = 4

Finding intersection of two tensors