add PixelUnShuffle layer

cszn · Mar 26, 2019 · 0a04dbe · 0a04dbe
1 parent b2f89d0
commit 0a04dbe
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -10,6 +10,8 @@
 # PixelUnshuffle layer (PyTorch)
 
 ``` python
+from torch.nn import Module
+
 
 def pixel_unshuffle(input, upscale_factor):
     r"""Rearranges elements in a Tensor of shape :math:`(C, rH, rW)` to a