Atletico Madrid scored three late goals to beat Real Sociedad 3-0, returning to winning ways and moving fourth in the La Liga standings.