当前位置: article > 正文

解决torch.nn.Embedding（index out of range in self）问题_return torch.embedding(weight, input, padding_idx,

作者：Monodyee | 2024-05-02 04:34:53

踩

return torch.embedding(weight, input, padding_idx, scale_grad_by_freq, spars

在使用torch时遇到Embedding的问题：return torch.embedding(weight, input, padding_idx, scale_grad_by_freq, sparse)

IndexError: index out of range in self

查了好多资料，看了官方文档都说是因为Embedding层张量输入超过范围，这个范围就是torch.nn.Embedding（num_embeddings,embedding_dim）中的num_embeddings，合法范围在[0,num_embeddings-1]内，对于输入的ini = torch.LongTensor([[]])数据不是看数据的个数是否超过范围，而是看数据的大小，通过打印

print(ini.min())

print(ini.max())

得到输入tensor的数据最大最小值判断是否超过合法范围

如果超过

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/Monodyee/article/detail/522204