假設有一個 4D 矩陣a=(20,3,100,100)。我想要分割的第一索引的a是20成4內的批次回圈，然后每次迭代追加到輸出。我不想使用 reshape，因為每個批次都將被送入網路。
輸出的大小應該是 (4,5,3,100,100)
這是我的嘗試：
output=[]
a=np.random.randint(0,100,size=(20,3,600,600))
for i in a[1:20:5,:,:,:] :
cnn_in=a[i,:,:,:]
output.append(cnn_in)
uj5u.com熱心網友回復：
像這樣的東西？
for i in range(0,20,5):
cnn_in=a[i:i 5]
output.append(cnn_in)
或更簡單的np.split：output=np.split(a,4)
轉載請註明出處，本文鏈接：https://www.uj5u.com/caozuo/340903.html
上一篇：如何通過將數字串列中的數字相加來找到給定的數字并回傳使用過的數字？
下一篇：填充由兩個串列組成的元組