perplexity = tensor(2.2968)