传统老鼠
np.where在多维数组的应用
函数用途返回查找的参数，在数组中的索引。Code举例：一般卷积神经网络的输入或者输出为一个四维的数组/Tensor。一般为[batch_size, channel, height, width]，下面代码目标是输出所有值为0的数字的索引。output = [[ [[1, 0, 2], [2, 1, 0], [1, 0, 0]]]]arr = np.array(output)print(arr.shape)res = np.where(arr==0)print(re
