diff --git a/imperative/python/megengine/functional/tensor.py b/imperative/python/megengine/functional/tensor.py
index 4cdebc155c27366d37ceb7b92a185bd0a72dbe29..8586f0b6991d234af5e367e1329977c22dbefa71 100644
--- a/imperative/python/megengine/functional/tensor.py
+++ b/imperative/python/megengine/functional/tensor.py
@@ -37,8 +37,7 @@ __all__ = [
     "broadcast",
     "concat",
     "cond_take",
-    "transpose",
-    "add_axis",
+    "dimshuffle",
     "eye",
     "flatten",
     "full",
@@ -732,7 +731,10 @@ def transpose(inp: Tensor, pattern: Iterable[int]) -> Tensor:
     return inp.transpose(pattern)
 
 
-dimshuffle = transpose
+def dimshuffle(inp: Tensor, pattern: Iterable[int]) -> Tensor:
+    r"""Same as :func:`~.transpose`.
+    """
+    return transpose(inp, pattern)
 
 
 def reshape(inp: Tensor, target_shape: Iterable[int]) -> Tensor: