fix(mge/functional): fix duplicate `transpose` in `__all__` and add `dimshuffle

GitOrigin-RevId: 2967088233

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: