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


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。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。