From b745342125e9f1eed4d94dbacc049e7297390108 Mon Sep 17 00:00:00 2001 From: SunAhong1993 Date: Thu, 3 Dec 2020 19:17:34 +0800 Subject: [PATCH] fix the reshape --- x2paddle/optimizer/tensorflow/prelu.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/x2paddle/optimizer/tensorflow/prelu.py b/x2paddle/optimizer/tensorflow/prelu.py index ee3d1e3..6bb8322 100644 --- a/x2paddle/optimizer/tensorflow/prelu.py +++ b/x2paddle/optimizer/tensorflow/prelu.py @@ -118,5 +118,6 @@ class PReLUOpt: graph.layers[transpose0.id] = transpose0 graph.layers[prelu.id] = prelu graph.layers[transpose1.id] = transpose1 - graph.parameters[alpha.outputs[0]] = np.reshape(graph.parameters[alpha.outputs[0]], (1, 0, 1, 1)) + axis1 = graph.parameters[alpha.outputs[0]].shape[0] + graph.parameters[alpha.outputs[0]] = np.reshape(graph.parameters[alpha.outputs[0]], (1, axis1, 1, 1)) graph.build() \ No newline at end of file -- GitLab