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


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