如何在torch.any中选择多个维度?

Question

我有一个(BxHxW)张量，它具有布尔值，可以在语义分割问题中理解语义映射。对于批处理中的每个语义映射，我需要了解该映射是否包含一些False。

我使用了torch.any(input, dim=(1, 2))，所以最后只剩下一个维度——男性的大小。但是，dim只能是int。

如何以最简洁的方式进行这样的检查，以及缺少这种看似简单的方法扩展的原因是什么。

Answer 1

您可以使用:

torch.any(input.view(input.shape[0], -1), dim=1)