diff --git a/imperative/python/megengine/functional/tensor.py b/imperative/python/megengine/functional/tensor.py index 76bd1055..3a3dbbc9 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