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


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