perplexity = tensor(5.5256)