- 博客(1)
- 收藏
- 关注
原创 基于pytorch,使用lstm网络实现变时长序列输入的分类任务
实现了使用lstm网络实现变时长序列输入的分类任务return padded_sequences.unsqueeze(2), labels # 增加一个维度作为 input_sizeoutputs, _ = model(sequences, torch.tensor([len(seq) for seq in sequences], dtype=torch.long).to(device)) # 确保长度张量也在设备上,并指定为 long 类型。
2024-07-02 19:10:30 220
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人