prelu 激活函数怎么用
收藏
RT 请教大神 在paddle中 prelu激活函数怎么用 act='prelu'显然是不行的
0
收藏
请登录后评论
import paddle.fluid as fluid
from paddle.fluid.dygraph.base import to_variable
import numpy as np
inp_np = np.ones([5, 200, 100, 100]).astype('float32')
with fluid.dygraph.guard():
inp_np = to_variable(inp_np)
mode = 'channel'
prelu = fluid.PRelu('prelu', mode=mode, param_attr=fluid.ParamAttr(initializer=fluid.initializer.Constant(1.0)))
dy_rlt = prelu(inp_np)
或者
import paddle.fluid as fluid
from paddle.fluid.param_attr import ParamAttr
x = fluid.data(name="x", shape=[None,5,10,10], dtype="float32")
mode = 'channel'
output = fluid.layers.prelu(
x,mode,param_attr=ParamAttr(name='alpha'))
我一般查文档,act参数里找不到,就去找PreLU对象,再没有就找fluid.layers下有没有。
这是1.8版本找法。2.0以后api重新布局了,更优美~~
查文档时可选不同版本paddle