我在运行pytorch的时候,出现了这样的错误,其实看到一些人又是加什么加什么。
我经过排查,最后发现只是超出了索引。
我本来的数据是torch.Size([32, 2, 20])
但是我却
data= data[:, [1, 0, 2], :]
我却让维度换了一下,我的数据只要两维,我换的时候却指定了三维,并且我是在gpu上运行的,就报了这个错误了。
其实只要注意维度和索引就可以了,我把数据改成torch.Size([32, 3, 20]),错误就解决了。
特别指出,果然只是在cpu上运行,那就是很正常的报错索引超出,但是在gpu运行的时候,吓死我了,我还以为出了什么大毛病呢