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

用法:

參數：

返回：

示例：