やっているときにこのエラーが発生する理由がわかりません:
batched_outputs = []
batched_outputs.append(output_sliced)
batched_outputs = torch.from_numpy(np.asarray(batched_outputs))
配列はどこoutput_sliced
にありますか。numpy
コードは正常に動作しているようです。おそらくあなたの問題output_sliced
ですか?
import numpy as np
import torch
batched_outputs = []
output_sliced = np.random.random((2,3))
batched_outputs.append(output_sliced)
batched_outputs = torch.from_numpy(np.asarray(batched_outputs))
上記のコードでエラーは発生しません。
これを意図しているかどうかはわかりませんが、次の点に注意してください。
batched_outputs = []
output_sliced = np.random.random((2,3))
batched_outputs.append(output_sliced)
print(np.asarray(batched_outputs).shape)
あなたにあげる
(1, 2, 3)