pixel_shuffle

    该OP将一个形为[N, C, H, W]的Tensor重新排列成形为 [N, C/r**2, H*r, W*r] 的Tensor。这样做有利于实现步长(stride)为1/r的高效sub-pixel(亚像素)卷积。详见Shi等人在2016年发表的论文 。

    参数:

    • 的多维Tensor,其中最后一维D是类别数目。数据类型为float32或float64。

    返回:根据新的维度信息进行重组的张量

    返回类型: Variable

    示例代码