From dbd07c73f5cb9d65de505c678d4a1845daa642f0 Mon Sep 17 00:00:00 2001
From: wujueying <wujueying1@huawei.com>
Date: Wed, 28 Dec 2022 16:18:49 +0800
Subject: [PATCH] fix Orgqr

---
 mindspore/core/ops/orgqr.cc | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/mindspore/core/ops/orgqr.cc b/mindspore/core/ops/orgqr.cc
index a740129c776..3dac37ca702 100644
--- a/mindspore/core/ops/orgqr.cc
+++ b/mindspore/core/ops/orgqr.cc
@@ -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 "