广播不同形状的阵列

Question

我有一个形状为（600,600,3）的图像，已使用形状（256,256）的窗口对其进行了分类，以使所得数组的形状为（400，256，256）。如何将其转换为（600,600）的形状？

我尝试过的是：

res= np.zeros((600,600))    
for im in pred:   # pred.shape = (400, 256, 256)
res+ = im

但这给了我一个错误：

  

ValueError：操作数不能与形状（600,600）（256,256）（600,600）一起广播