You've got exactly the right image in mind: higher partial pressure of a gas in the lungs corresponds to a higher dissolved concentration of that species (H2, He, N2, O2, whatever) in the blood and tissue.