完成argmax公式验证并修改Bug