fix TensorPy init empty Tensor (#1962)

mindspore-lab · Feb 19, 2025 · 3ea3f4a · 3ea3f4a
1 parent d859a1b
commit 3ea3f4a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/mindnlp/core/ops/creation.py b/mindnlp/core/ops/creation.py
@@ -123,7 +123,7 @@ def empty(*size, dtype=None):
         size = size[0]
     if dtype is None:
         dtype = get_default_dtype()
-    out = CTensor(dtype, size)
+    out = CTensor(dtype=dtype, shape=size)
     return mindspore.Tensor(out)
 
 # empty_like

diff --git a/mindnlp/transformers/models/rag/modeling_rag.py b/mindnlp/transformers/models/rag/modeling_rag.py
@@ -1064,8 +1064,7 @@ def marginalize(self, seq_logits, doc_scores, n_docs=None):
         )
         doc_logprobs = F.log_softmax(doc_scores, dim=1)
         log_prob_sum = seq_logprobs + doc_logprobs.unsqueeze(-1).unsqueeze(-1)
-        # return ops.logsumexp(log_prob_sum, dim=1)
-        return mindspore.ops.logsumexp(log_prob_sum, axis=1)
+        return ops.logsumexp(log_prob_sum, dim=1)
 
     def forward(
             self,