Constant¶
该接口为常量初始化函数,用于权重初始化,通过输入的 value 值初始化输入变量。
参数¶
value (float16|float32,可选) - 用于初始化输入变量的值,默认值为 0。
返回¶
参数初始化类的实例。
代码示例¶
import paddle
import paddle.nn as nn
data = paddle.rand([30, 10, 2], dtype='float32')
linear = nn.Linear(2,
4,
weight_attr=nn.initializer.Constant(value=2.0))
res = linear(data)
print(linear.weight)
# Tensor(shape=[2, 4], dtype=float32, place=Place(gpu:0), stop_gradient=False,
# [[2., 2., 2., 2.],
# [2., 2., 2., 2.]])