weixin_44578857的博客

原创 RuntimeError ‘lengths‘ argument should be a 1D CPU int64 tensor, but got 1D cuda0 Long tensor问题解决方法

RuntimeError: ‘lengths’ argument should be a 1D CPU int64 tensor, but got 1D cuda:0 Long tensor问题解决方法在使用rnn网络的pack_padded_sequence出现RuntimeError: ‘lengths’ argument should be a 1D CPU int64 tensor, but got 1D cuda:0 Long tensor问题，可使用如下解决方式：将nn.utils.rnn