p83571920
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!21252 fix bug of unique 

Merge pull request !21252 from lianliguang/fix-bug-of-unique
This commit is contained in:
i-robot 2021-08-03 07:49:17 +00:00 committed by Gitee
parent 9aa662d0fc b7bcb2ee29
commit bb0728d25b
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/core/abstract/prim_arrays.cc
View File

@ -140,7 +140,7 @@ AbstractBasePtr InferImplUnique(const AnalysisEnginePtr &, const PrimitivePtr &p
auto shape = input->shape();
MS_EXCEPTION_IF_NULL(shape);
if (shape->shape().empty()) {
if (shape->shape().size() != 1) {
MS_LOG(EXCEPTION) << "Rank of " << op_name << "'s input must be 1.";
}
ShapeVector ids_shape = {Shape::SHP_ANY};