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


Python mxnet.module.BaseModule.backward用法及代码示例


用法:

backward(out_grads=None)

参数

out_grads(NDArray or list of NDArray, optional) - 要传播返回的输出的梯度。仅当在不是损失函数的输出上调用 bind 时才需要此参数。

后向计算。

例子

>>> # An example of backward computation.
>>> mod.backward()
>>> print mod.get_input_grads()[0].asnumpy()
[[[  1.10182791e-05   5.12257748e-06   4.01927764e-06   8.32566820e-06
    -1.59775993e-06   7.24269375e-06   7.28067835e-06  -1.65902311e-05
     5.46342608e-06   8.44196393e-07]
     ...]]

相关用法


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