From 71a927c03d85bcd1ca4ae304f7ddc3ed9d3950dc Mon Sep 17 00:00:00 2001
From: zhaojiaying01 <zhaojiaying01@baidu.com>
Date: Tue, 13 Aug 2019 20:05:14 +0800
Subject: [PATCH] fix scale input's dims not equal output's dims

---
 src/operators/kernel/arm/scale_kernel.cpp | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/operators/kernel/arm/scale_kernel.cpp b/src/operators/kernel/arm/scale_kernel.cpp
index 92fffeaed7..fffcb07533 100644
--- a/src/operators/kernel/arm/scale_kernel.cpp
+++ b/src/operators/kernel/arm/scale_kernel.cpp
@@ -31,6 +31,9 @@ template <>
 void ScaleKernel<CPU, float>::Compute(const ScaleParam<CPU> &param) {
   const auto input = param.InputX();
   auto output = param.Out();
+  if (input->dims() != output->dims()) {
+    output->Resize(input->dims());
+  }
   const float scale = param.Scale();
   const float bias = param.Bias();
   if (input->type() == type_id<int64_t>().hash_code()) {
-- 
GitLab