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

用法:

參數：

返回：

示例：