提交 3cf49cbf 编写于 作者: S SunAhong1993

fix the expand_dim for numpy 1.17

上级 0b0335f9
...@@ -118,5 +118,7 @@ class PReLUOpt: ...@@ -118,5 +118,7 @@ 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.expand_dims(graph.parameters[alpha.outputs[0]], axis=(0, 2, 3)) 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.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.
先完成此消息的编辑!
想要评论请 注册