x = token_embedding + position_embedding