![](/img/trans.png)
[英]Numpy Error: can't convert np.ndarray of type numpy.object_
[英]TypeError: can't convert np.ndarray of type numpy.object_
如何将dtype=object
的 numpy 数组转换为 torch Tensor
?
array([
array([0.5, 1.0, 2.0], dtype=float16),
array([4.0, 6.0, 8.0], dtype=float16)
], dtype=object)
很难正确回答,因为您没有向我们展示您如何尝试这样做。 从您的错误消息中,我可以看到您尝试将包含对象的 numpy 数组转换为火炬张量。 这不起作用,您将需要一个数字数据类型:
import torch
import numpy as np
# Your test array without 'dtype=object'
a = np.array([
np.array([0.5, 1.0, 2.0], dtype=np.float16),
np.array([4.0, 6.0, 8.0], dtype=np.float16),
])
b = torch.from_numpy(a)
print(a.dtype) # This should not be 'object'
print(b)
输出
float16
tensor([[0.5000, 1.0000, 2.0000],
[4.0000, 6.0000, 8.0000]], dtype=torch.float16)
只需添加上面写的内容-
首先,您应该确保您的数组 dtype 不是“O”(对象)。
你这样做:( 信用)
a=np.vstack(a).astype(np.float)
然后你可以使用:
b = torch.from_numpy(a)
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.