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
示例代码
本文档使用 BookStack 构建