当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python mxnet.gluon.Parameter.initialize用法及代码示例


用法:

initialize(init=None, ctx=None, default_init=<mxnet.initializer.Uniform object>, force_reinit=False)

参数

  • init(Initializer) - 要使用的初始化程序。覆盖Parameter.init()和default_init。
  • ctx(上下文或上下文列表,默认为context.current_context().) -在给定的上下文中初始化参数。如果 ctx 是 Context 列表,将为每个 context 制作一份副本。

    注意

    副本是独立的数组。用户有责任在更新时保持其值一致。通常 gluon.Trainer 会为您执行此操作。

  • default_init(Initializer) - 两者都使用默认初始化器init()Parameter.init()None.
  • force_reinit(bool, default False) - 如果参数已经初始化,是否强制重新初始化。

初始化参数和梯度数组。仅用于NDArray API。

例子

>>> weight = mx.gluon.Parameter('weight', shape=(2, 2))
>>> weight.initialize(ctx=mx.cpu(0))
>>> weight.data()
[[-0.01068833  0.01729892]
 [ 0.02042518 -0.01618656]]
<NDArray 2x2 @cpu(0)>
>>> weight.grad()
[[ 0.  0.]
 [ 0.  0.]]
<NDArray 2x2 @cpu(0)>
>>> weight.initialize(ctx=[mx.gpu(0), mx.gpu(1)])
>>> weight.data(mx.gpu(0))
[[-0.00873779 -0.02834515]
 [ 0.05484822 -0.06206018]]
<NDArray 2x2 @gpu(0)>
>>> weight.data(mx.gpu(1))
[[-0.00873779 -0.02834515]
 [ 0.05484822 -0.06206018]]
<NDArray 2x2 @gpu(1)>

相关用法


注:本文由纯净天空筛选整理自apache.org大神的英文原创作品 mxnet.gluon.Parameter.initialize。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。