`tf.nn.space_to_depth` 是否有等效的 PyTorch 函数

【问题标题】：Is there an equivalent PyTorch function for `tf.nn.space_to_depth``tf.nn.space_to_depth` 是否有等效的 PyTorch 函数
【发布时间】：2020-03-10 11:13:00
【问题描述】：

正如标题所说，tf.nn.space_to_depth 是否有等效的 PyTorch 函数？

【问题讨论】：

标签： python tensorflow pytorch

【解决方案1】：

虽然torch.nn.functional.pixel_shuffle 的功能与tf.nn.depth_to_space 完全相同，但PyTorch 没有任何功能来执行类似于tf.nn.space_to_depth 的逆运算。

话虽如此，使用torch.nn.functional.unfold 很容易实现space_to_depth。

def space_to_depth(x, block_size):
    n, c, h, w = x.size()
    unfolded_x = torch.nn.functional.unfold(x, block_size, stride=block_size)
    return unfolded_x.view(n, c * block_size ** 2, h // block_size, w // block_size)

【讨论】：

仅供参考：pytorch 现在提供类似于tf.space_to_depth 的pixel-unshuffle