fix the reshape

18cc4a40 · SunAhong1993 · b7453421 · 18cc4a40
显示空白变更内容
内联并排

Showing with 2 addition and 2 deletion

x2paddle/optimizer/tensorflow/prelu.py x2paddle/optimizer/tensorflow/prelu.py +2 -2

未找到文件。
--- a/x2paddle/optimizer/tensorflow/prelu.py
+++ b/x2paddle/optimizer/tensorflow/prelu.py
@@ -118,6 +118,6 @@ class PReLUOpt:
                graph.layers[transpose0.id] = transpose0
                graph.layers[prelu.id] = prelu
                graph.layers[transpose1.id] = transpose1
-                axis1 = graph.parameters[alpha.outputs[0]].shape[0]
+                first_axis = graph.parameters[alpha.outputs[0]].shape[0]
-                graph.parameters[alpha.outputs[0]] = np.reshape(graph.parameters[alpha.outputs[0]], (1, axis1, 1, 1))
+                graph.parameters[alpha.outputs[0]] = np.reshape(graph.parameters[alpha.outputs[0]], (1, first_axis, 1, 1))
        graph.build()
\ No newline at end of file