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


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