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

Merge pull request #408 from Channingss/add_prelu

[paddle2onnx] add prelu
...@@ -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.
先完成此消息的编辑!
想要评论请 注册