perplexity = tensor(2.5999)