perplexity = tensor(3.3904)