當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文


Python mxnet.executor.Executor.reshape用法及代碼示例


用法:

reshape(partial_shaping=False, allow_up_sizing=False, **kwargs)

參數

  • partial_shaping(bool) - 是否允許更改未指定參數的形狀。
  • allow_up_sizing(bool) - 是否允許分配比原來更大的新 ndarray。
  • kwargs(dict of string to tuple of int) - 參數的新形狀。

返回

exec- 一個與自己共享內存的新執行器。

返回類型

mxnet.executor.Executor

返回具有相同符號和共享內存但輸入/輸出形狀不同的新執行器。對於運行時整形、變長序列等。返回的執行器與當前執行器共享狀態,不能與其並行使用。

例子

>>> a = mx.sym.Variable('a')
>>> b = mx.sym.Variable('b')
>>> c = 2 * a + b
>>> texec = c.bind(mx.cpu(), {'a': mx.nd.zeros((2, 1)), 'b': mx.nd.ones((2,1))})
>>> new_shape = {'a': (4, 2), 'b': (4, 2)}
>>> texec.reshape(allow_up_sizing=True, **new_shape)

相關用法


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