Numpy 错误：无法转换 numpy.object_ 类型的 np.ndarray

Question

不知道为什么我在做的时候会收到这个错误：

batched_outputs = []
batched_outputs.append(output_sliced)
batched_outputs = torch.from_numpy(np.asarray(batched_outputs))

其中output_sliced是一个numpy数组。

Answer 1

代码似乎工作正常。 也许你的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)