Uniform

class paddle.nn.initializer. Uniform ( low=- 1.0, high=1.0, name=None ) [源代码]

随机均匀分布初始化函数。

参数:

  • low (float,可选) - 下界。默认值为 -1.0。

  • high (float,可选) - 上界。默认值为 1.0。

  • name (str,可选)- 具体用法请参见 Name,一般无需设置,默认值为None。

返回:

由随机均匀分布初始化的参数。

代码示例

  1. import paddle
  2. data = paddle.ones(shape=[3, 1, 2], dtype='float32')
  3. weight_attr = paddle.framework.ParamAttr(
  4. name="linear_weight",
  5. initializer=paddle.nn.initializer.Uniform(low=-0.5, high=0.5))
  6. bias_attr = paddle.framework.ParamAttr(
  7. name="linear_bias",
  8. initializer=paddle.nn.initializer.Uniform(low=-0.5, high=0.5))
  9. linear = paddle.nn.Linear(2, 2, weight_attr=weight_attr, bias_attr=bias_attr)
  10. # linear.weight: [[-0.46245047 0.05260676]
  11. # [ 0.38054508 0.29169726]]
  12. # linear.bias: [-0.2734719 0.23939109]
  13. res = linear(data)
  14. # res: [[[-0.3553773 0.5836951]]
  15. # [[-0.3553773 0.5836951]]
  16. # [[-0.3553773 0.5836951]]]