Predict token from positional embedding