diff --git a/imperative/python/megengine/functional/nn.py b/imperative/python/megengine/functional/nn.py index b0e4f52e..025b30ea 100644 --- a/imperative/python/megengine/functional/nn.py +++ b/imperative/python/megengine/functional/nn.py @@ -78,6 +78,7 @@ __all__ = [ "max_pool2d", "one_hot", "prelu", + "pad", "relu", "relu6", "remap",