Browse Source

feat(imperative): add PixelShuffle to module.__init__.py

release-1.7
chenjiahui 3 years ago
parent
commit
c8a9a3fd0a
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      imperative/python/megengine/module/__init__.py

+ 1
- 0
imperative/python/megengine/module/__init__.py View File

@@ -33,6 +33,7 @@ from .lrn import LocalResponseNorm
from .module import Module from .module import Module
from .normalization import GroupNorm, InstanceNorm, LayerNorm from .normalization import GroupNorm, InstanceNorm, LayerNorm
from .padding import Pad from .padding import Pad
from .pixel_shuffle import PixelShuffle
from .pooling import AvgPool2d, MaxPool2d from .pooling import AvgPool2d, MaxPool2d
from .quant_dequant import DequantStub, QuantStub from .quant_dequant import DequantStub, QuantStub
from .sequential import Sequential from .sequential import Sequential


Loading…
Cancel
Save