问RuntimeError：“长度”参数应该是一个一维CPU int64张量，但是得到了一维cuda:0长张量。
EN

Stack Overflow用户

提问于 2021-12-20 21:16:11

回答 1查看 2.7K关注 0票数 3

我试着做一个文本分类，使用电筒和火炬文本在纸空间。

我得到了

RuntimeError: ‘lengths’ argument should be a 1D CPU int64 tensor, but got 1D cuda:0 Long tensor

我的PyTorch版本是1.10.1+cu102

发布于 2021-12-20 22:28:14

昨天我遇到了这个问题，在我的例子中，rnn序列希望长度在cpu上，所以在函数调用中将长度放在CPU上，如下所示：

packed_sequences = nn.utils.rnn.pack_padded_sequence(padded_tensor, valid_frames.to('cpu'), batch_first=True, enforce_sorted=True)

这可能不是您所使用的确切函数，但我认为它将适用于大多数functions函数。

票数 4

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/70428140

复制

相似问题

问RuntimeError：“长度”参数应该是一个一维CPU int64张量，但是得到了一维cuda:0长张量。EN