Python mxnet.symbol.Symbol.infer_shape_partial用法及代碼示例

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

參數：

*args：- 以位置方式顯示參數的形狀。未知形狀可以標記為無
**kwargs：- 已知形狀的關鍵字參數。

arg_shapes：(list of tuple or None) - 參數形狀列表。順序與list_arguments()的順序相同。
out_shapes：(list of tuple or None) - 輸出形狀列表。順序與list_outputs()的順序相同。
aux_shapes：(list of tuple or None) - 輔助狀態形狀列表。順序與list_auxiliary_states()的順序相同。

部分推斷形狀。

此函數的工作方式與 infer_shape 相同，隻是此函數可以返回部分結果。

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

示例：

>>> data = mx.sym.Variable('data')
>>> prev = mx.sym.Variable('prev')
>>> fc1  = mx.sym.FullyConnected(data=data, name='fc1', num_hidden=128)
>>> fc2  = mx.sym.FullyConnected(data=prev, name='fc2', num_hidden=128)
>>> out  = mx.sym.Activation(data=mx.sym.elemwise_add(fc1, fc2), act_type='relu')
>>> out.list_arguments()
['data', 'fc1_weight', 'fc1_bias', 'prev', 'fc2_weight', 'fc2_bias']
>>> out.infer_shape(data=(10,64))
(None, None, None)
>>> out.infer_shape_partial(data=(10,64))
([(10L, 64L), (128L, 64L), (128L,), (), (), ()], [(10L, 128L)], [])
>>> # infers shape if you give information about fc2
>>> out.infer_shape(data=(10,64), prev=(10,128))
([(10L, 64L), (128L, 64L), (128L,), (10L, 128L), (128L, 128L), (128L,)], [(10L, 128L)], [])

相關用法

注：本文由純淨天空篩選整理自apache.org大神的英文原創作品 mxnet.symbol.Symbol.infer_shape_partial。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。

用法:

參數：

返回：

示例：