perplexity = tensor(3.5858)