numpy 有一个方法，你可以用 np.all

There is a way in numpy , you can do with np.all

a = np.random.rand(3, 1, 2)
b = a[1][0]
np.all(np.all(a == b, 1), 1)
Out[612]: array([False,  True, False])

来自bnaecker的解决方案

Solution from bnaecker

np.all(a == b, axis=(1, 2))

如果只想检查是否退出

np.any(np.all(a == b, axis=(1, 2)))