From 20332705d3daa154f4f33b0ad381e92031317a93 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Wed, 8 Apr 2020 21:57:17 +0800 Subject: [PATCH] docs(mge/functional): refine tensor transpose doc GitOrigin-RevId: 763d74d61e5aaf502267f7b16348d480aceb0406 --- python_module/megengine/functional/tensor.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/python_module/megengine/functional/tensor.py b/python_module/megengine/functional/tensor.py index 270604cc0..e1a7322c0 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 -- GitLab