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


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


用法:

get_outputs(merge_multi_context=True)

参数

merge_multi_context(bool) - 默认为True.在使用data-parallelism 的情况下,将从多个设备收集输出。一种Truevalue 表示我们应该合并收集的结果,以便它们看起来像来自单个执行器。

返回

输出

返回类型

NDArray 列表或 NDArray 列表列表。

获取上一个前向计算的输出。

如果 merge_multi_contextTrue ,则类似于 [out1, out2] 。否则,它以 [[out1_dev1, out1_dev2], [out2_dev1, out2_dev2]] 形式返回输出。所有输出元素的类型均为 NDArray 。当 merge_multi_contextFalse 时,那些 NDArray 实例可能存在于不同的设备上。

例子

>>> # An example of getting forward output.
>>> print mod.get_outputs()[0].asnumpy()
[[ 0.09999977  0.10000153  0.10000716  0.10000195  0.09999853  0.09999743
   0.10000272  0.10000113  0.09999088  0.09999888]]

相关用法


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