错误处理:IndexError: index out of range in self
生活随笔
收集整理的這篇文章主要介紹了
错误处理:IndexError: index out of range in self
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
import torch
t_embeddings = torch.nn.Embedding(2,5)
x=torch.tensor([[2,3]])
y=t_embeddings(x)
代碼如上,報錯如下:
IndexError: index out of range in self????????報這個錯誤是embedding層的張量輸入超過了合法范圍,embedding層的合法張量輸入數值范圍應該在[0, num_embeddings - 1]的范圍內,過大過小都會報錯。
????????因此可以在embedding層先輸出查看一下最大范圍:?
? ? ? ? ?比如下面這樣,就沒有問題:
t_embeddings = torch.nn.Embedding(5,2) x=torch.tensor([[2,3]]) y=t_embeddings(x) y ''' tensor([[[ 1.7594, -0.2992],[-0.1612, 0.9231]]], grad_fn=<EmbeddingBackward>) '''總結
以上是生活随笔為你收集整理的错误处理:IndexError: index out of range in self的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: python 笔记 haversine
- 下一篇: 反距离插值 IDW