Constant

class paddle.nn.initializer. Constant ( value=0.0 ) [源代码]

该接口为常量初始化函数,用于权重初始化,通过输入的 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.]])

使用本API的教程文档