提交 422412a0 编写于 作者: C channingss

[paddle2onnx] add prelu

上级 2c12e94b
...@@ -110,6 +110,13 @@ class OpSet9(object): ...@@ -110,6 +110,13 @@ class OpSet9(object):
'Relu', inputs=op.input('X'), outputs=op.output('Out')) 'Relu', inputs=op.input('X'), outputs=op.output('Out'))
return node return node
def prelu(self, op, block):
node = helper.make_node(
'PRelu',
inputs=[op.input('X')[0], op.input('Alpha')[0]],
outputs=op.output('Out'))
return node
def tanh(self, op, block): def tanh(self, op, block):
node = helper.make_node( node = helper.make_node(
'Tanh', inputs=op.input('X'), outputs=op.output('Out')) 'Tanh', inputs=op.input('X'), outputs=op.output('Out'))
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册