fix ComputeAccidentalHits example

pull/10107/head
yanzhenxiang2020 4 years ago
parent a350613b72
commit 29bb99d55c

@ -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};

@ -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]
"""

Loading…
Cancel
Save