Python mxnet.test_utils.check_symbolic_forward用法及代碼示例

用法: mxnet.test_utils.check_symbolic_forward(sym, location, expected, rtol=None, atol=None, aux_states=None, ctx=None, equal_nan=False, dtype=<class 'numpy.float32'>)

參數：

sym：(Symbol) - 輸出符號
location：(list of np.ndarray or dict of str to np.ndarray) -評價點
- 如果類型是np.ndarray 的列表
  包含與 sym.list_arguments() 對應的所有 numpy 數組。
- 如果類型是 str 到 np.ndarray 的字典
  包含參數名稱及其值之間的映射。
expected：(list of np.ndarray or dict of str to np.ndarray) -預期輸出值
- 如果類型是np.ndarray 的列表
  包含對應於exe.outputs的數組
- 如果類型是 str 到 np.ndarray 的字典
  包含 sym.list_output() 和 exe.outputs 之間的映射
rtol：(None or float) - 相對閾值。如果設置為，將使用默認閾值None.
atol：(None or float) - 絕對閾值。如果設置為，將使用默認閾值None.
aux_states：(list of np.ndarray of dict, optional) -
- 如果類型是np.ndarray 的列表
  包含與 sym.list_auxiliary_states 對應的所有 NumPy 數組
- 如果類型是 str 到 np.ndarray 的字典
  包含輔助狀態名稱及其值之間的映射。
ctx：(mxnet.context.Context, optional) - 運行上下文
dtype：("asnumpy" or np.float16 or np.float32 or np.float64) - 如果 dtype 是 “asnumpy” 那麽創建的 mx.nd.array 將具有與從中複製它的 numpy 數組相同的類型。否則，dtype 是在此函數中創建的所有 mx.nd.array 對象的顯式數據類型。
equal_nan：(Boolean) - 如果是真的，nan是檢查等效性的有效值(即nan==nan)

將交易品種的前向結果與預期結果進行比較。如果轉發結果與預期結果不同，則打印錯誤消息。

示例：

>>> shape = (2, 2)
>>> lhs = mx.symbol.Variable('lhs')
>>> rhs = mx.symbol.Variable('rhs')
>>> sym_dot = mx.symbol.dot(lhs, rhs)
>>> mat1 = np.array([[1, 2], [3, 4]])
>>> mat2 = np.array([[5, 6], [7, 8]])
>>> ret_expected = np.array([[19, 22], [43, 50]])
>>> check_symbolic_forward(sym_dot, [mat1, mat2], [ret_expected])

相關用法

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