From 9c5ea68992e5ebc0174a7a3b81f3a8919f30a490 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Fri, 9 Oct 2020 22:59:00 +0800 Subject: [PATCH] feat(mge): remove dimshuffle GitOrigin-RevId: 9e327ee3afd1325bae70f2f3f84cdfd28a5e9ed9 --- imperative/python/megengine/functional/tensor.py | 7 ------- 1 file changed, 7 deletions(-) diff --git a/imperative/python/megengine/functional/tensor.py b/imperative/python/megengine/functional/tensor.py index 76bd10552..3a3dbbc9e 100644 --- a/imperative/python/megengine/functional/tensor.py +++ b/imperative/python/megengine/functional/tensor.py @@ -36,7 +36,6 @@ __all__ = [ "broadcast", "concat", "cond_take", - "dimshuffle", "expand_dims", "eye", "flatten", @@ -730,12 +729,6 @@ def transpose(inp: Tensor, pattern: Iterable[int]) -> Tensor: return inp.transpose(pattern) -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: r""" Reshapes a tensor to given target shape; total number of logical elements must -- GitLab