diff --git a/x2paddle/op_mapper/caffe_custom_layer/normalize.py b/x2paddle/op_mapper/caffe_custom_layer/normalize.py index 577a674f7e70aebee78735611c11e7202b2ccf7e..3cef4e6551a2656a0da656bf5330deffa111a061 100644 --- a/x2paddle/op_mapper/caffe_custom_layer/normalize.py +++ b/x2paddle/op_mapper/caffe_custom_layer/normalize.py @@ -15,7 +15,7 @@ def normalize_layer(inputs, input = inputs[0] l2_norm = fluid.layers.l2_normalize(input, axis=1, name=name + '_l2') scale_param = fluid.layers.create_parameter( - shape=[1] if channel_shared else [input_shape[0][1]], + shape=[1] if channel_shared else [1, 1, 1, input_shape[0][1]], dtype=input.dtype, attr=name + '_scale') scale_param = fluid.layers.reshape(x=scale_param, \