未验证 提交 52752736 编写于 作者: J Jason 提交者: GitHub

Merge pull request #448 from SunAhong1993/paddle-2.0

fix the reshape
...@@ -118,5 +118,6 @@ class PReLUOpt: ...@@ -118,5 +118,6 @@ class PReLUOpt:
graph.layers[transpose0.id] = transpose0 graph.layers[transpose0.id] = transpose0
graph.layers[prelu.id] = prelu graph.layers[prelu.id] = prelu
graph.layers[transpose1.id] = transpose1 graph.layers[transpose1.id] = transpose1
graph.parameters[alpha.outputs[0]] = np.reshape(graph.parameters[alpha.outputs[0]], (1, 0, 1, 1)) first_axis = graph.parameters[alpha.outputs[0]].shape[0]
graph.parameters[alpha.outputs[0]] = np.reshape(graph.parameters[alpha.outputs[0]], (1, first_axis, 1, 1))
graph.build() graph.build()
\ No newline at end of file
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册