Python mxnet.symbol.Symbol.infer_type用法及代码示例

用法:
infer_type(*args, **kwargs)

参数：

*args：- 以位置方式的已知参数类型。未知类型可以标记为无。
**kwargs：- 已知类型的关键字参数。

arg_types：(list of numpy.dtype or None) - 参数类型列表。顺序与list_arguments()的顺序相同。
out_types：(list of numpy.dtype or None) - 输出类型列表。顺序与list_outputs()的顺序相同。
aux_types：(list of numpy.dtype or None) - 辅助状态类型列表。顺序与list_auxiliary_states()的顺序相同。

给定某些参数的已知类型，推断所有参数和所有输出的类型。

此函数以位置方式或关键字参数方式将某些参数的已知类型作为输入。如果没有足够的信息来推断缺失的类型，它会返回一个 None 值的元组。

已知类型的不一致将导致引发错误。

示例：

>>> a = mx.sym.var('a')
>>> b = mx.sym.var('b')
>>> c = a + b
>>> arg_types, out_types, aux_types = c.infer_type(a='float32')
>>> arg_types
[<type 'numpy.float32'>, <type 'numpy.float32'>]
>>> out_types
[<type 'numpy.float32'>]
>>> aux_types
[]

相关用法

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

用法:

参数：

返回：

示例：