perplexity = tensor(3.5966)