📝 Pixel Shuffle 与 Pixel UnShuffle

1. pixel_shuffle

• 假设输入张量的形状为 (batch_size, channels, height, width)，并且我们希望使用因子 r 进行 pixel_shuffle。

• 输入的 channels 必须是 r^2 的倍数，即 channels = new_channels * r^2。

• 操作将输入张量重新排列为形状 (batch_size, new_channels, height * r, width * r) 的输出张量。

2. pixel_unshuffle

• 假设输入张量的形状为 (batch_size, channels, height, width)，并且我们希望使用因子 r 进行 pixel_unshuffle。

• 输入的 height 和 width 必须是 r 的倍数，即 height = new_height * r 和 width = new_width * r。

• 操作将输入张量重新排列为形状 (batch_size, channels * r^2, height / r, width / r) 的输出张量。

总结

• pixel_shuffle：用于上采样，将通道维度转换为空间维度，提高图像的分辨率。主要用于生成更高分辨率的图像。

• pixel_unshuffle：用于下采样，将空间维度转换为通道维度，降低图像的分辨率。主要用于生成更低分辨率的图像。

🤗 总结归纳