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


Python mxnet.symbol.Symbol.list_auxiliary_states用法及代碼示例


用法:

list_auxiliary_states()

返回

aux_states- 輸入符號中的輔助狀態列表。

返回類型

str 列表

列出符號中的所有輔助狀態。

示例

>>> a = mx.sym.var('a')
>>> b = mx.sym.var('b')
>>> c = a + b
>>> c.list_auxiliary_states()
[]

BatchNorm 中的輔助狀態示例。

>>> data = mx.symbol.Variable('data')
>>> weight = mx.sym.Variable(name='fc1_weight')
>>> fc1  = mx.symbol.FullyConnected(data = data, weight=weight, name='fc1', num_hidden=128)
>>> fc2 = mx.symbol.BatchNorm(fc1, name='batchnorm0')
>>> fc2.list_auxiliary_states()
['batchnorm0_moving_mean', 'batchnorm0_moving_var']

注意

輔助狀態是符號的特殊狀態,與參數不對應,不通過梯度下降進行更新。輔助狀態的常見示例包括 BatchNorm 中的 moving_meanmoving_variance。大多數運算符沒有輔助狀態。

相關用法


注:本文由純淨天空篩選整理自apache.org大神的英文原創作品 mxnet.symbol.Symbol.list_auxiliary_states。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。