PyTorch：两个二进制掩码联合？

Question

我有两个形状为(batch_size, width, heigh) 的二进制掩码，我想创建一个二进制掩码来指示两者之间元素的联合。

要找到交集，我可以使用torch.where(A == B, 1, 0)，但是如何找到并集呢？

Answer 1

使用二进制掩码时，应使用逻辑运算，例如：
logical_or()、logical_and()。

那么交集就是二进制掩码：

intersection = A.logical_and(B)

联合是：

union = A.logical_or(B)

---

顺便说一句，
我将把它留给您作为练习，以检查您计算的交集 (A == B) 不正确的原因。