Update loss.py

11e9d4e3 · HydrogenSulfate · chajchaj · 0c2d6bcb · 11e9d4e3
显示空白变更内容
内联并排

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
@@ -1806,8 +1806,8 @@ def cross_entropy(input,

            valid_label = paddle.where(label == ignore_index,
                                       paddle.zeros_like(label), label)
-            ignore_weight_mask = paddle.cast(
-                (label != ignore_index), input.dtype)
+            ignore_weight_mask = paddle.cast((label != ignore_index),
+                                             input.dtype)
            if ignore_weight_mask.ndim > 1 and ignore_weight_mask.shape[
                    -1] == 1:
                ignore_weight_mask = paddle.squeeze(ignore_weight_mask, -1)