[Bug fixed] Change squeeze() to squeeze(0) to accomodate for sequence length of 1#1581

ChenWu98 wants to merge 1 commit intoThilinaRajapakse:masterThilinaRajapakse/simpletransformers:masterfrom

ChenWu98:masterChenWu98/simpletransformers:masterCopy head branch name to clipboard