Merge pull request #5803 from WenmuZhou/pse

rm view
2026-09-24 23:33:08 +08:00 · 2022-03-28 15:15:57 +08:00
parent e55a72e099 fef47dcc18
commit 4348fc35f7
1 changed files with 3 additions and 3 deletions
@@ -121,9 +121,9 @@ class PSELoss(nn.Layer):

        if neg_num == 0:
            selected_mask = training_mask
-            selected_mask = selected_mask.view(
-                1, selected_mask.shape[0],
-                selected_mask.shape[1]).astype('float32')
+            selected_mask = selected_mask.reshape(
+                [1, selected_mask.shape[0], selected_mask.shape[1]]).astype(
+                    'float32')
            return selected_mask

        neg_score = paddle.masked_select(score, gt_text <= 0.5)