pgjmblvch
/
mindspore2022
forked from huawei/mindspore2022
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix ComputeAccidentalHits example

This commit is contained in:
yanzhenxiang2020 2020-12-17 16:26:59 +08:00
parent a350613b72
commit 29bb99d55c
2 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -529,7 +529,7 @@ AbstractBasePtr InferImplComputeAccidentalHits(const AnalysisEnginePtr &, const
auto shape = input->shape();
if (shape->shape().size() != 2) {
MS_LOG(EXCEPTION) << "Rank of " << op_name << "'s input must be 1.";
MS_LOG(EXCEPTION) << "Rank of " << op_name << "'s input must be 2.";
}
ShapeVector indices_shape = {Shape::SHP_ANY};
ShapeVector min_shape = {1};

5
mindspore/ops/operations/nn_ops.py
View File

@ -3425,8 +3425,9 @@ class ComputeAccidentalHits(PrimitiveWithCheck):
>>> sampler = ops.ComputeAccidentalHits(2)
>>> output1, output2, output3 = sampler(Tensor(x), Tensor(y))
>>> print(output1, output2, output3)
[0, 0, 1, 1, 2, 2], [1, 2, 0, 4, 3, 3],
[-3.4028235+38, -3.4028235+38, -3.4028235+38, -3.4028235+38, -3.4028235+38, -3.4028235+38]
[0 0 1 1 2 2]
[1 2 0 4 3 3]
[-3.4028235e+38 -3.4028235e+38 -3.4028235e+38 -3.4028235e+38 -3.4028235e+38 -3.4028235e+38]
"""