Arbitrary polarization changing devices are described within paraxial approximation by their action on the spatial angular Stokes parameters associated with the generalized rays of a Wigner formulation of wave optics. This approach leads to the introduction of generalized Mueller matrices being functions of the spatial and angular variables parameterizing generalized rays. We apply this approach to a polarization grating.