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


Python mxnet.symbol.Variable用法及代码示例


用法:

mxnet.symbol.Variable(name, attr=None, shape=None, lr_mult=None, wd_mult=None, dtype=None, init=None, stype=None, **kwargs)

参数

  • name(str) - 变量名。
  • attr(Dict of strings) - 要在变量上设置的附加属性。格式{字符串:字符串}。
  • shape(tuple) - 变量的形状。如果指定,这将在形状推断期间使用。如果在调用形状推断时使用关键字参数为该变量指定了不同的形状,则该形状信息将被忽略。
  • lr_mult(float) - 输入变量的学习率乘数。
  • wd_mult(float) - 输入变量的权重衰减乘数。
  • dtype(str or numpy.dtype) - 输入变量的 dtype。如果未指定,将推断此值。
  • init(initializer (mxnet.init.*)) - 此变量的初始化程序(可选)覆盖默认初始化程序。
  • stype(str) - 变量的存储类型,如‘row_sparse’, ‘csr’、‘default’等
  • kwargs(Additional attribute variables) - 附加属性必须以双下划线开头和结尾。

返回

variable- 对应于计算图输入的符号。

返回类型

Symbol

创建具有指定名称的符号变量。

示例

>>> data = mx.sym.Variable('data', attr={'a': 'b'})
>>> data
<Symbol data>
>>> csr_data = mx.sym.Variable('csr_data', stype='csr')
>>> csr_data
<Symbol csr_data>
>>> row_sparse_weight = mx.sym.Variable('weight', stype='row_sparse')
>>> row_sparse_weight
<Symbol weight>

相关用法


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