pytorch embedding层报错index out of range in self

最新推荐文章于 2024-04-14 16:59:35 发布

嗷海胆

最新推荐文章于 2024-04-14 16:59:35 发布

阅读量2.6w

点赞数 26

本文链接：https://blog.csdn.net/qq_32743513/article/details/109959952

版权

在使用 PyTorch 的 Embedding 层时，遇到`index out of range in self`错误。问题源于输入数据未经词典映射，导致数值超出了有效范围[0, num_embeddings-1]。通过检查张量的最小值和最大值，修正超出范围的数值，可以避免此错误。提供了一个修正后的正确使用embedding层的示例。" 132134546,11389520,Python实现心电信号时域特征分析,"['Python', '开发语言', '数据分析', '生物医学信号处理', '心电图分析']

摘要由CSDN通过智能技术生成

使用pytorch时，数据过embedding层时报错：

Traceback (most recent call last):
  File "C:/Users/gaosiqi/PycharmProjects/DeepFM/main.py", line 68, in <module>
    out = model(train_data)
  File "C:\Anaconda3\envs\tensorflow\lib\site-packages\torch\nn\modules\module.py", line 722, in _call_impl
    result = self.forward(*input, **kwargs)
  File "C:/Users/gaosiqi/PycharmProjects/DeepFM/main.py", line 26, in forward
    embedding = self.word_embedding(x)
  File "C:\Anaconda3\envs\tensorflow\lib\site-packages\torch\nn\modules\module.py", line 722, in _call_impl
    result = self.forward(*input, **kwargs)
  File "C:\Anaconda3\envs\tensorflow\lib\site-packages\torch\nn\modules\sparse.py", line 126, in forward
    self.norm_type, self.scale_grad_by_freq, self.sparse)
  File "C:\Anaconda3\envs\tensorflow\lib\site-packages\torch\nn\functional.py", line 1814, in embedding
    return torch.embedding(weight, inp