Diego Costa scored on his La Liga return for Atletico Madrid in a 2-0 win over Getafe - but his joy was short-lived as he received a second yellow card for his celebrations.