mindspore重写WithLossCell,TrainOneStepCell接口
错误:RuntimeError: _kernel.cc:88 CheckParam] AddN output shape must be equal to input shape.Trace: In file add_impl.py(272)/ return F.addn((x, y))/
解决方案:不要在WithLossCell的construct方法中返回多个参数,否则就会在TrainOneStepCell的construct中的GradOperation就会报错。