python/paddle/fluid/layers/loss.py · ff8252387e5039f4bbc1201da38a7a956a562669 · 机器未来 / Paddle

[NPU] use SparseSoftmaxCrossEntropyWithLogits in npu kernel of softmax_with_cross_entropy (#32858) · ff825238

由 Leo Chen 提交于 6月 15, 2021

* use SparseSoftmaxCrossEntropyWithLogits

* fix

* test_slice

* revert test_slice

* add backprob for npu kernel

* fix typo

* fix ut

* fix ut

* refine comments

* return softmax

ff825238

loss.py 72.2 KB

机器未来 / Paddle 与 Fork 源项目一致

Replace loss.py

机器未来 / Paddle
与 Fork 源项目一致