我得到了一个大小为 (30, 256, 256) 的 ML 算法分割掩码张量。问题是，其中一些条目是零矩阵，我必须将它们过滤掉。现在我使用基于 for 循环的技术和 np.array_equal 函数来手动过滤掉它们。
有没有办法使用一些奇特的索引以 NumPy 方式更有效地完成此操作？
最佳答案
如果a是你的数组，你可以这样做
a[np.any(a != 0, (1, 2))]
关于python - 从numpy中的张量中过滤出零矩阵，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/51459156/