From 328ded50765ba0f6b0253f06fa97b06b57f148da Mon Sep 17 00:00:00 2001 From: SunAhong1993 Date: Thu, 3 Dec 2020 18:00:53 +0800 Subject: [PATCH] modify --- x2paddle/optimizer/tensorflow/prelu.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/x2paddle/optimizer/tensorflow/prelu.py b/x2paddle/optimizer/tensorflow/prelu.py index c920605..ee3d1e3 100644 --- a/x2paddle/optimizer/tensorflow/prelu.py +++ b/x2paddle/optimizer/tensorflow/prelu.py @@ -118,7 +118,5 @@ class PReLUOpt: graph.layers[transpose0.id] = transpose0 graph.layers[prelu.id] = prelu graph.layers[transpose1.id] = transpose1 - graph.parameters[alpha.outputs[0]] = np.expand_dims(graph.parameters[alpha.outputs[0]], axis=(0)) - graph.parameters[alpha.outputs[0]] = np.expand_dims(graph.parameters[alpha.outputs[0]], axis=(2)) - graph.parameters[alpha.outputs[0]] = np.expand_dims(graph.parameters[alpha.outputs[0]], axis=(3)) + graph.parameters[alpha.outputs[0]] = np.reshape(graph.parameters[alpha.outputs[0]], (1, 0, 1, 1)) graph.build() \ No newline at end of file -- GitLab