Expected tensor for argument #1 ‘indices‘ to have scalar type Long； but got torch.cuda.FloatTensor i

最新推荐文章于 2024-07-27 15:52:39 发布

欣欣然xl

最新推荐文章于 2024-07-27 15:52:39 发布

阅读量5.6k

点赞数 3

分类专栏： pytorch 机器学习文章标签： cuda pytorch gpu

本文链接：https://blog.youkuaiyun.com/weixin_37817539/article/details/114537927

版权

当遇到`Expected tensor for argument #1 ‘indices‘ to have scalar type Long`错误时，通常是由于nn.Embedding输入不是torch.LongTensor。解决办法是在输入前转换为torch.LongTensor或指定dtype为int64。了解torch中tensor数据类型与Python数据类型的对应关系有助于避免这类问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Expected tensor for argument #1 'indices' to have scalar type Long; but got torch.cuda.FloatTensor instead (while checking arguments for embedding)

我遇到这个问题时是由于我nn.Embedding()的输入并非torch.int64(torch.LongTensor)，解决方法是使用torch.LongTensor()或者是dtype=int64，这是证明

torch中tensor的数据类型和python中的数据类型的对照如下：

data type	CPU type	GPU type
32-bit floating point	torch.FloatTensor	torch.cuda.FloatTensor
64-bit floating point	torch.DoubleTensor	torch.cuda.DoubleTensor
16-bit floating point	N/A	torch.cuda.HalfTensor
8-bit integer (unsigned)	torch.ByteTenso