The world number four was heckled after throwing his racquet in the third set of his loss to Stan Wawrinka