From f0172aee4ca8f8ed433f3d6e0ba0d22ca210d02b Mon Sep 17 00:00:00 2001
From: fuzhiye <fuzhiye@huawei.com>
Date: Mon, 10 Aug 2020 15:33:18 +0800
Subject: [PATCH] fix bug of reshape infershape

---
 mindspore/lite/src/ops/reshape.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mindspore/lite/src/ops/reshape.cc b/mindspore/lite/src/ops/reshape.cc
index 7285de3877..bd30a74ad1 100644
--- a/mindspore/lite/src/ops/reshape.cc
+++ b/mindspore/lite/src/ops/reshape.cc
@@ -92,7 +92,7 @@ int Reshape::InferShape(std::vector<tensor::Tensor *> inputs_, std::vector<tenso
       MS_LOG(INFO) << "Do infer shape in runtime.";
       return RET_INFER_INVALID;
     }
-    size_t shape_size = shape_tensor->shape().size();
+    size_t shape_size = shape_tensor->ElementsNum();
     switch (shape_tensor->data_type()) {
       case kNumberTypeInt8: {
         auto data = reinterpret_cast<int8_t *>(shape_tensor->Data());