提交 225ec4e0 编写于 作者: Z zhaojiaying01

fix scale input's dims not equal output's dims

上级 a61c31df
......@@ -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()) {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册