当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


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。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。