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

Python mxnet.util.set_np_shape用法及代码示例


用法:

mxnet.util.set_np_shape(active)

参数：

active：(bool) - 指示是否打开/关闭NumPy 形状语义。

返回：

返回类型：

一个布尔值，指示 NumPy 形状语义的先前状态。

打开/关闭NumPy形状语义，其中()表示标量张量的形状，带有0元素的元组，例如(0,)、(1, 0, 2)表示zero-size张量的形状。默认情况下关闭此函数以保持向后兼容性。

请注意，这是为传入的MXNet-NumPy 运算符设计的基础设施。在模块mx.nd 和mx.sym 中注册的传统运算符在此语义中不能保证其行为与NumPy 中的对应项相同。

示例：

>>> import mxnet as mx
>>> prev_state = mx.set_np_shape(True)
>>> print(prev_state)
False
>>> print(mx.is_np_shape())
True

相关用法


注：本文由纯净天空筛选整理自apache.org大神的英文原创作品 mxnet.util.set_np_shape。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。