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


Python mxnet.symbol.var用法及代碼示例

用法:

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