fix softmax_cross_entropy return val (#50218)

0673a54f · YuhangLi · GitHub · 32e977c5 · 0673a54f
隐藏空白更改
内联并排

Showing with 2 addition and 2 deletion

python/paddle/nn/functional/loss.py python/paddle/nn/functional/loss.py +2 -2

未找到文件。
--- a/python/paddle/nn/functional/loss.py
+++ b/python/paddle/nn/functional/loss.py
@@ -2601,7 +2601,7 @@ def cross_entropy(
            "npu"
        ) or core.is_compiled_with_custom_device("mlu"):
            if not soft_label:
-                _, _, out = _legacy_C_ops.softmax_with_cross_entropy(
+                _, out = _legacy_C_ops.softmax_with_cross_entropy(
                    input,
                    valid_label,
                    'soft_label',
@@ -2616,7 +2616,7 @@ def cross_entropy(
                    use_softmax,
                )
            else:
-                _, _, out = _legacy_C_ops.softmax_with_cross_entropy(
+                _, out = _legacy_C_ops.softmax_with_cross_entropy(
                    input,
                    label,
                    'soft_label',