pytorch embedding 报错

使用pytorch写代码时候，报这个错误

…/aten/src/ATen/native/cuda/Indexing.cu:1146: indexSelectLargeIndex: block: [116,0,0], thread: [73,0,0] Assertion srcIndex < srcSelectDimSize` failed.

经查，我的是因为 nn.Embedding 的问题。因为我申请词表的范围没有包括填充的字符。所以遇到填充字符就会报错。

这个如果不处理，虽然能让程序继续往下运行，但是当经过cuda操作（例如 前向网络）就会报错。所以必须修复。