perplexity = tensor(1.3368)