perplexity = tensor(2.0204)