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


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


用法:

simple_bind(ctx, grad_req='write', type_dict=None, stype_dict=None, group2ctx=None, shared_arg_names=None, shared_exec=None, shared_buffer=None, **kwargs)

參數

  • ctx(mxnet.context.Context) - 生成的執行程序運行的設備上下文。
  • grad_req(string) -{‘write’, ‘add’, ‘null’},或 str 列表或 str 到 str 的字典,可選指定我們應該如何將梯度更新為 args_grad
    • 'write' 表示每次將梯度寫入指定的args_grad NDArray。
    • 'add' 表示每次將梯度添加到指定的 NDArray。
    • 'null' 表示不采取任何行動,可能不計算梯度。
  • type_dict(Dict of str->numpy.dtype) - 輸入類型字典,name->dtype
  • stype_dict(Dict of str->str) - 輸入存儲類型字典,名稱->storage_type
  • group2ctx(Dict of string to mx.Context) - 字典映射ctx_group歸因於上下文分配。
  • shared_arg_names(List of string) - 參數名稱NDArrayshared_exec 的一部分可用於初始化當前執行程序。
  • shared_exec(mxnet.executor.Executor) - 其arg_arrays、arg_arrays、grad_arrays和aux_arrays可以重用於初始化當前執行器的執行器。
  • shared_buffer(字符串字典到NDArray) - dict 映射參數名稱到NDArray可以重用於初始化當前執行器。如果當前執行程序的一個參數名稱在shared_arg_names.這NDArrays 應具有默認存儲類型。
  • kwargs(Dict of str->shape) - 輸入形狀字典,名稱->形狀

返回

executor- 生成的執行器

返回類型

mxnet.Executor

綁定當前符號以獲取執行器,分配所需的所有參數。允許指定數據類型。

此函數簡化了綁定過程。您隻需要指定輸入數據形狀。在綁定執行程序之前,該函數分配未明確指定的參數和輔助狀態。允許指定數據類型。

示例

>>> x = mx.sym.Variable('x')
>>> y = mx.sym.FullyConnected(x, num_hidden=4)
>>> exe = y.simple_bind(mx.cpu(), x=(5,4), grad_req='null')
>>> exe.forward()
[<NDArray 5x4 @cpu(0)>]
>>> exe.outputs[0].asnumpy()
array([[ 0.,  0.,  0.,  0.],
       [ 0.,  0.,  0.,  0.],
       [ 0.,  0.,  0.,  0.],
       [ 0.,  0.,  0.,  0.],
       [ 0.,  0.,  0.,  0.]], dtype=float32)
>>> exe.arg_arrays
[<NDArray 5x4 @cpu(0)>, <NDArray 4x4 @cpu(0)>, <NDArray 4 @cpu(0)>]
>>> exe.grad_arrays
[<NDArray 5x4 @cpu(0)>, <NDArray 4x4 @cpu(0)>, <NDArray 4 @cpu(0)>]

相關用法


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