python - 是否有一个函数可以找到 pytorch 中 bool 矩阵中每一行的 OR？

Question

我有一个大小为 n x m 的矩阵 A，所有条目都是 bool 值。我希望我的所有计算都在 GPU 上进行，并且我将矩阵 A 存储为张量，每个条目都是 pytorch 的 bool 数据类型。我希望单个向量 b 的输出是一个 1 x m 张量，用于存储 A 中所有行的 OR。

我想要的是:
矩阵 =
[a1,1, a1,2, , a1,3, ... , a1,m]
[a2,1, a2,2, , a2,3, ... , a2,m]
...
[an,1, an,2, , an,3, ... , an,m]

b = [b1, b2, , b3, , ..., bm]

英石。 bi = a1,i | a2,i | a3,i | ... |一个，任何地方|是 Pytorch 中的 OR 运算符

基本上我想要一个应用行或列明智 bool 运算的函数。我知道| .__OR__可以使用，并且 Pytorch 的 OR 函数采用两个 bool 张量作为输入，我需要遍历所有行以获得所需的 b 向量。
由于 OR 是可交换的 (a|(b|c) = (a|b)|c)，我认为 pytorch 会有一些不错的功能，可以通过执行 | 来加速它并行或以某种分而治之的方法进行操作，而不是使用循环来执行此操作。欢迎任何想法或引用来加快使用 pytorch 应用可通信的行/列 bool 运算的过程。最好是所有操作都在 GPU 上完成。

Answer 1

见 torch.any 和 torch.all .

两者都拿了 dim参数，因此您可以计算 or/and行。