remove squeeze to avoid changing 1D-tensor to scalar (#749)

joey12300 · web-flow · commit 5afc790de234 · 2021-07-14T17:24:43.000+08:00
diff --git a/paddlenlp/layers/crf.py b/paddlenlp/layers/crf.py
@@ -145,7 +145,7 @@ def forward(self, inputs, lengths):
         if self.with_start_stop_tag:
             # The last one step
             alpha += self.transitions[self.stop_idx].unsqueeze(0)
-        norm_score = log_sum_exp(alpha, 1).squeeze(-1)
+        norm_score = log_sum_exp(alpha, 1)  #.squeeze(-1)
         return norm_score
 
     def gold_score(self, inputs, labels, lengths):
