Код работает нормально. Возможно проблема с вашим 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)