Miscellaneous Layers#

Module

MAdds

FLOPs

Parameters

Memory

torch.nn.ChannelShuffle

x

x

✓

x

torch.nn.PixelShuffle

x

x

✓

x

torch.nn.PixelUnshuffle

x

x

✓

x

torch.nn.Upsample

x

✓

✓

x

torch.nn.UpsamplingNearest2d

x

x

✓

x

torch.nn.UpsamplingBilinear2d

x

x

✓

x