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

fix Orgqr

This commit is contained in:
wujueying 2022-12-28 16:18:49 +08:00
parent 3ccff77d3b
commit dbd07c73f5
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/core/ops/orgqr.cc
View File

@ -52,6 +52,10 @@ abstract::ShapePtr OrgqrInferShape(const PrimitivePtr &, const std::vector<Abstr
<< ", while x.shape[-2] is " << x_shape[rank - kRowIndex] << " and x.shape[-1] is "
<< x_shape[rank - kColIndex] << ".";
}
if (tau_shape.size() < 1) {
MS_EXCEPTION(ValueError) << "For Orgqr, tau should have one dimension less than x"
<< ", but tensor has no dimensions";
}
if (*(x_shape.end() - 1) < *(tau_shape.end() - 1)) {
MS_EXCEPTION(ValueError) << "For Orgqr, x.shape[-1] must be greater than or equal to tau.shape[-1]"
<< ", while x.shape[-1] is " << x_shape[rank - kColIndex] << " and "