diff --git a/python_module/megengine/functional/tensor.py b/python_module/megengine/functional/tensor.py
index 270604cc0d468b220b0f1dc94a3e5b33825ac65f..e1a7322c06943b20e9ddf4504cc747696417499b 100644
--- a/python_module/megengine/functional/tensor.py
+++ b/python_module/megengine/functional/tensor.py
@@ -445,11 +445,10 @@ def reshape(inp: Tensor, target_shape: Iterable[int]) -> Tensor:
     return mgb.opr.reshape(inp, target_shape)
 
 
-@functools.wraps(dimshuffle)
-def transpose(*args, **kwargs):
-    r"""See :func:`dimshuffle`
+def transpose(inp: Tensor, pattern: Iterable[int]) -> Tensor:
+    r"""Equivalent to :func:`dimshuffle`
     """
-    return dimshuffle(*args, **kwargs)
+    return dimshuffle(inp, pattern)
 
 
 @wrap_io_tensor