Gradient estimates for the heat equation under the Ricci flow