小编J. *_*nou的帖子

我得到了一个 3 维形状 (n,m,k) numpy 数组。我想将其视为包含向量的二维矩阵，即具有大小为 k 的向量的 nxm 矩阵。我现在想检查两个这样的形状数组 (n,m,k) 是否第一个数组中的条目 (x,y,:) 等于第二个数组中的 (x,y,:) 。有没有一种方法可以在不使用循环的情况下在 numpy 中做到这一点？

我想过类似 A == B 以第一和第二轴为条件的东西。