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


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


用法:

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

參數

  • data(Symbol) - 要壓縮的數據
  • axis(Shape or None, optional, default=None) - 選擇形狀中一維條目的子集。如果選擇了形狀條目大於 1 的軸,則會引發錯誤。
  • name(string, optional.) - 結果符號的名稱。

返回

結果符號。

返回類型

Symbol

從數組的形狀中刪除一維條目。在大多數情況下,將輸出張量形狀定義為 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.symbol.op.squeeze。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。