numpy 数组逻辑运算用于索引

注意事项

numpy数组逻辑运算用于索引时，会将数组拉平
例：

a = np.ones((3, 26, 26))
b = np.random.rand(3, 26, 26)
n = b[a == 1]
c = (a == 1)
print(n.shape)
print(c.shape)

运行结果：

(2028,)
(3, 26, 26)

作为笔记