For an incompressible flow , but with variable density, Chia-Shun Yih derived the necessary equation. The velocity field is first transformed using Yih transformation
where  is some reference density, with corresponding Stokes streamfunction  defined such that
Let us include the gravitational force acting in the negative  direction. The Yih equation is then given by
where