提交 b7453421 编写于 作者: S SunAhong1993

fix the reshape

上级 328ded50
...@@ -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)) 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() 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.
先完成此消息的编辑!
想要评论请 注册