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


Python mxnet.ndarray.op.squeeze用法及代碼示例


用法:

mxnet.ndarray.op.squeeze(data=None, axis=_Null, out=None, name=None, **kwargs)

參數

  • data(NDArray) - 要壓縮的數據
  • axis(Shape or None, optional, default=None) - 選擇形狀中一維條目的子集。如果選擇了形狀條目大於 1 的軸,則會引發錯誤。
  • out(NDArray, optional) - 輸出 NDArray 來保存結果。

返回

out- 此函數的輸出。

返回類型

NDArray 或 NDArray 列表

從數組的形狀中刪除一維條目。在大多數情況下,將輸出張量形狀定義為 numpy.squeeze 的行為相同。有關例外情況,請參閱以下注釋。例子:

data = [[[0], [1], [2]]]
squeeze(data) = [0, 1, 2]
squeeze(data, axis=0) = [[0], [1], [2]]
squeeze(data, axis=2) = [[0, 1, 2]]
squeeze(data, axis=(0, 2)) = [0, 1, 2]

注意

此運算符的輸出將保持至少一個維度不被刪除。例如,squeeze([[[4]]]) = [4],而在numpy.squeeze 中,輸出將變為標量。

相關用法


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