现在在使用pytorch中的Embedding层的时候出现了index out of range in self的报错,报错信息如下,现在记录一下解决方法。
发现这个问题主要是由self.entity_embeddings(LongTensor[head]))这一条语句引起的,我的解决方法是把它换成self.entity_embeddings(Variable(LongTensor([head])))
换了之后就不报错了,具体原因还没找到。
index out of range in self 的解决方法
最新推荐文章于 2024-09-19 15:42:45 发布