rm view

2026-09-24 23:33:08 +08:00 · 2022-03-28 05:05:07 +00:00
parent d9f1fed7a8
commit fef47dcc18
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)