perplexity = tensor(2.2976)