Browse Source

fix argmaxwithvalue error in pynative mode

tags/v0.6.0-beta
chujinjin 5 years ago
parent
commit
1663a92f83
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/ccsrc/backend/session/session_basic.cc

+ 1
- 1
mindspore/ccsrc/backend/session/session_basic.cc View File

@@ -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) {


Loading…
Cancel
Save