perplexity = tensor(3.5839)