perplexity = tensor(2.2887)