fix argmaxwithvalue error in pynative mode

5 years ago · 1663a92f83
parent 45ad430af2
commit 1663a92f83
1 changed files with 1 additions and 1 deletions
--- a/mindspore/ccsrc/backend/session/session_basic.cc
+++ b/mindspore/ccsrc/backend/session/session_basic.cc
@ -482,7 +482,7 @@ CNodePtr SessionBasic::CreateNewCNode(const CNodePtr &cnode, bool valid_input, K
        cnode_inputs.emplace_back(new_value_node);
      }
      continue;
-    } else if (anf->isa<Parameter>()) {
+    } else if (anf->isa<Parameter>() && AnfAlgo::GetOutputTensorNum(anf) == 1) {
      auto new_parameter = CreateNewParameterFromParameter(anf, valid_input, graph);
      cnode_inputs.push_back(new_parameter);
      if (GetGraphIdByNode(anf) == kInvalidGraphId) {