ValueError:expected sequence of length 10 at dim 1 (got 1)

错误现象:

数据预处理创建mini batch时,因为以下代码导致出错:ValueError:expected sequence of length 10 at dim 1 (got 1)

    inout_seq.append((train_seq, train_label))
    return torch.FloatTensor(inout_seq)

原因分析:

train_seq和 train_label 长度一不一样,一个有10个元素,另一个只有一个。

解决方案:

修改好的办法是, 在划分trainset, testset的时候, 就把数据转换成torch的tensor:

    train_data = df[:samples]   
    test_data = df[samples:]
    
    train_data = torch.FloatTensor(train_data).to(device)
    test_data = torch.FloatTensor(test_data).to(device)

这样就可以直接

return inout_seq

顺利解决error

附:

x = [[1,2,3,4], [2]]
x = torch.FloatTensor(x)
print(x)

输出ValueError: expected sequence of length 4 at dim 1 (got 1)

改为:

x = [[1,2,3,4], [2,3,4,5]]
x = torch.FloatTensor(x)
print(x)

就可以正常输出:

tensor([[1., 2., 3., 4.],
        [2., 3., 4., 5.]])

总结:

内部机制要求两个列表维度一样。

是否有自动补全不确定,没有深入研究。

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

DLANDML

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值