Momentum diffusion of atoms moving in laser fields