Browse Source

feat(mge): remove dimshuffle

GitOrigin-RevId: 9e327ee3af
release-1.1
Megvii Engine Team 4 years ago
parent
commit
9c5ea68992
1 changed files with 0 additions and 7 deletions
  1. +0
    -7
      imperative/python/megengine/functional/tensor.py

+ 0
- 7
imperative/python/megengine/functional/tensor.py View File

@@ -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


Loading…
Cancel
Save