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

Python mxnet.gluon.contrib.nn.PixelShuffle1D用法及代碼示例

用法:

class mxnet.gluon.contrib.nn.PixelShuffle1D(factor)

參數：

  • factor：(int or 1-tuple of int) - 上采樣因子，應用於W維度。
  • Inputs：-
    • data: 形狀張量(N, f*C, W).
  • Outputs：-
    • out: 形狀張量(N, C, W*f).

基礎：mxnet.gluon.block.HybridBlock

Pixel-shuffle 用於一維上采樣的層。

Pixel-shuffling 是沿channel 維度獲取值組並將它們沿W 維度重新組合成像素塊的操作，從而有效地將該維度乘以一個常數因子。

例如，通過形成大小為 的小值組並將它們排列在大小為 的網格中，將形狀為 的特征圖重塑為 。

例子：

>>> pxshuf = PixelShuffle1D(2)
>>> x = mx.nd.zeros((1, 8, 3))
>>> pxshuf(x).shape
(1, 4, 6)

相關用法


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