From 0673a54fedb9b3ab50b1abd583e08bc6af1d3d43 Mon Sep 17 00:00:00 2001
From: YuhangLi <104877312+piDack@users.noreply.github.com>
Date: Mon, 6 Feb 2023 10:37:50 +0800
Subject: [PATCH] fix softmax_cross_entropy return val (#50218)

---
 python/paddle/nn/functional/loss.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/python/paddle/nn/functional/loss.py b/python/paddle/nn/functional/loss.py
index a441183ca82..9930b09780d 100644
--- 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',
-- 
GitLab