From c8a9a3fd0a6995732984e68492bcdb155ad0b3a7 Mon Sep 17 00:00:00 2001 From: chenjiahui Date: Thu, 21 Oct 2021 18:28:57 +0800 Subject: [PATCH] feat(imperative): add PixelShuffle to module.__init__.py --- imperative/python/megengine/module/__init__.py | 1 + 1 file changed, 1 insertion(+) diff --git a/imperative/python/megengine/module/__init__.py b/imperative/python/megengine/module/__init__.py index 3e127fa7..73b12474 100644 --- a/imperative/python/megengine/module/__init__.py +++ b/imperative/python/megengine/module/__init__.py @@ -33,6 +33,7 @@ from .lrn import LocalResponseNorm from .module import Module from .normalization import GroupNorm, InstanceNorm, LayerNorm from .padding import Pad +from .pixel_shuffle import PixelShuffle from .pooling import AvgPool2d, MaxPool2d from .quant_dequant import DequantStub, QuantStub from .sequential import Sequential