複数画像でチャネルの合計値を作りたい

Question

CNNの中間にあるConv層を抜いたチャネルが2048枚、形状は（7,7,2048）になります。全チャネルの合計値を計算する際、1画像ならできるのですが、複数枚（例、8画像（8,7,7,2048））になると方法がわかりません。

データ形状とコード

print(layer_outputs.shape) #（8,7,7,2048）

#形状指定後、h,wをゼロでパディングします。
h, w, ch = layer_outputs.shape[1:]
res = np.zeros((h,w))

#1画像なら下記で抽出できます
for i in range(ch):
    img_res = layer_outputs[0,:,:,i] #0番目
    res = res + img_res 

#複数画像での試行
lst = []

for i in range(ch):
    for n in range(len(layer_outputs)):
        img_res = layer_outputs[n,:,:,i]
        res = res + img_res
        lst.append(res)

#結果は16384(8*2048)と画像数別になりません。
np.array(lst).shape #(16384, 7, 7)

他にも

img_res = layer_outputs[n,:,:,i]を
img_res = layer_outputs[n][:,:,i]
で試行してみましたが、結果は上記と同じでした。

希望する形状は（8,7,7）になります。
どのようにすれば、画像別に合計値を算出できるのでしょうか。
ご教示頂ければ幸甚です。よろしくお願いします。

Answer 1

わかりました。2段階に分ければ入りました。お騒がせしました。

lst=[]

for img_idx in range(len(X)):
    layer_output = layer_outputs[img_idx]
    h, w, ch = layer_output.shape
    res = np.zeros((h,w))
    
    for i in range(ch):
        img_res = layer_output[:,:,i]
        res = res + img_res

    lst.append(res)