diff --git a/src/operators/kernel/arm/scale_kernel.cpp b/src/operators/kernel/arm/scale_kernel.cpp index 92fffeaed74c4286f7e32153b8356952a4de1639..fffcb07533490e490e0e62044d477c8ff54089d5 100644 --- a/src/operators/kernel/arm/scale_kernel.cpp +++ b/src/operators/kernel/arm/scale_kernel.cpp @@ -31,6 +31,9 @@ template <> void ScaleKernel::Compute(const ScaleParam ¶m) { 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().hash_code()) {