simplify loss
simplify loss

cihangxie/cleverhans,carlini/cleverhans,carlini/cleverhans,cleverhans-lab/cleverhans,openai/cleverhans,fartashf/cleverhans,cleverhans-lab/cleverhans,cleverhans-lab/cleverhans