當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文


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。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。