TypeError: can‘t convert np.ndarray of type numpy.object_. The only supported types are: float64, fl

文章讲述了在使用Python的PyTorch库时,遇到numpy数组元素为object类型导致无法转换为Tensor的问题。解决方法是将numpy数组强制转换为PyTorch支持的数据类型,如float64,float32等。
摘要由CSDN通过智能技术生成
def get_loader(data, data_normal, seq_length, u_num, batch_size):
    x = []
    for i in range(len(data_normal) - seq_length - 1):
        _x = data_normal[i : i + seq_length]
        print(_x)
        print('*' * 50)
        # print(_u)
        x.append(_x)
    x = np.array(x)
    x_data = Variable(torch.Tensor(np.array(x)))
    dataset = Data.TensorDataset(x_data)
    loader = torch.utils.data.DataLoader(dataset=dataset, batch_size=batch_size, shuffle=False, drop_last=True)
    return loader

seq_length = 10 
batch_size = 16
u_num = 2 

n = 100
data_normal = torch.randn((n, 1, 4))
train_loader = get_loader(data1, data_normal, seq_length, u_num, batch_size)


报错:

1. VisibleDeprecationWarning: Creating an ndarray from ragged nested sequences (which is a list-or-tuple of lists-or-tuples-or ndarrays with different lengths or shapes) is deprecated. If you meant to do this, you must specify 'dtype=object' when creating the ndarray.

2. TypeError: can't convert np.ndarray of type numpy.object_. The only supported types are: float64, float32, float16, complex64, complex128, int64, int32, int16, int8, uint8, and bool.

错误原因:

1. 

读入的numpy数组里的元素是object类型,无法将这种类型转换成tensor。

解决方法:将numpy数组进行强制类型转换成float类型(或者任何pytorch支持的类型:float64, float32, float16, int64, int32, int16, int8, uint8, and bool)即可。

  • 6
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值