Fix token size overflow

[kyakuno]

tokensがpush_backされた結果、attention_maskよりもtoken.sizeのほうが大きくなり、エラーになる問題を修正。