in many places it does things such as
deprel_seqs = [self.vocab['deprel'].unmap([preds[1][i][j+1][h] for j, h in enumerate(hs)]) for i, hs in enumerate(head_seqs)]
which, while unlikely, includes PAD and the other vocab prefix options as legal results