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

Question

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

要找到交集，我可以使用torch.where(A == B, 1, 0)，但我怎样才能找到并集？

标签： pytorch

Answer 1

使用二进制掩码时，您应该使用逻辑运算，例如：
logical_or(), logical_and().

交集是二进制掩码：

intersection = A.logical_and(B)

工会是：

union = A.logical_or(B)

---

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