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

用法:
infer_type_partial(*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()的顺序相同。

部分推断类型。

此函数的工作方式与 infer_type 相同，只是此函数可以返回部分结果。

在以下示例中，有关 fc2 的信息不可用。因此，infer_shape 将返回 None 值的元组，但 infer_shape_partial 将返回部分值。

示例：

>>> data = mx.sym.Variable('data')
>>> prev = mx.sym.Variable('prev')
>>> casted_prev  = mx.sym.cast(prev, dtype='float32')
>>> out  = mx.sym.Activation(data=mx.sym.elemwise_add(data, casted_prev), act_type='relu')
>>> out.list_arguments()
['data', 'prev']
>>> out.infer_type(data='float32')
(None, None, None)
>>> out.infer_type_partial(data='float32')
([numpy.float32, None], [numpy.float32], [])
>>> # infers type if you give information about prev
>>> out.infer_type(data='float32', prev='float16')
([numpy.float32, numpy.float16], [numpy.float32], [])

相关用法

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

用法:

参数：

返回：

示例：