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

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