remove erroneous comment in ArgMaxLayer