我想迭代形狀為 (1134500, 1, 50) 的 3d 陣列(序列)
array([[[1000, 1000, 1000, ..., 1005, 1005, 1005]],
[[1000, 1000, 1000, ..., 1004, 1005, 1004]],
[[1000, 1000, 1000, ..., 1004, 1005, 1004]],
...,
[[1000, 1000, 1000, ..., 1005, 1005, 1004]],
[[1000, 1000, 1000, ..., 1005, 1005, 1005]],
[[1000, 1000, 1000, ..., 1004, 1005, 1004]]], dtype=int32)
為此,我使用了以下 for 回圈,它運行良好,除了覆寫之前批處理的結果:
batchsize = 500
for i in range(0, sequences.shape[0], batchsize):
batch = sequences[i:i batchsize]
relevances = lrp_model.lrp(batch)
結果,我想要一個形狀為 (1134500, 1, 50) 的陣列(相關性),但我得到一個形狀為 (500, 1, 50) 的陣列(相關性)有人能告訴我出了什么問題嗎?
uj5u.com熱心網友回復:
如果您想保存相關性,也許
batchsize = 500
relevances = np.zeros(sequences.shape)
for i in range(0, sequences.shape[0], batchsize):
batch = sequences[i:i batchsize]
relevances[i:i batchsize, :, :] = lrp_model.lrp(batch)
轉載請註明出處,本文鏈接:https://www.uj5u.com/qita/371796.html
上一篇:如何將物件陣列轉換為鍵值對
