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

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。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。