From 1fab8908c66757bdcc596556c8605572336409ca Mon Sep 17 00:00:00 2001 From: jiaqianjing Date: Wed, 21 Sep 2022 15:40:58 +0800 Subject: [PATCH] update for ranking_loss #46302 (#46306) --- python/paddle/nn/functional/loss.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/python/paddle/nn/functional/loss.py b/python/paddle/nn/functional/loss.py index 3f5637fa39..f1cc6c1147 100755 --- a/python/paddle/nn/functional/loss.py +++ b/python/paddle/nn/functional/loss.py @@ -1159,8 +1159,9 @@ def margin_ranking_loss(input, check_variable_and_dtype(label, 'label', ['float32', 'float64'], 'margin_rank_loss') - out = paddle.subtract(other, input) - out = paddle.multiply(out, label) + out = paddle.subtract(input, other) + neg_label = paddle.neg(label) + out = paddle.multiply(neg_label, out) if margin != 0.0: margin_var = out.block.create_var(dtype=out.dtype) -- GitLab