perplexity = tensor(3.3638)