Embedding Error Index out of Range in self

@ptrblck can you please shed some light on it https://discuss.pytorch.org/t/embedding-layer/121969