提交 9c5ea689 编写于 作者: M Megvii Engine Team

feat(mge): remove dimshuffle

GitOrigin-RevId: 9e327ee3afd1325bae70f2f3f84cdfd28a5e9ed9
上级 1e2117f6
...@@ -36,7 +36,6 @@ __all__ = [ ...@@ -36,7 +36,6 @@ __all__ = [
"broadcast", "broadcast",
"concat", "concat",
"cond_take", "cond_take",
"dimshuffle",
"expand_dims", "expand_dims",
"eye", "eye",
"flatten", "flatten",
...@@ -730,12 +729,6 @@ def transpose(inp: Tensor, pattern: Iterable[int]) -> Tensor: ...@@ -730,12 +729,6 @@ def transpose(inp: Tensor, pattern: Iterable[int]) -> Tensor:
return inp.transpose(pattern) 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: def reshape(inp: Tensor, target_shape: Iterable[int]) -> Tensor:
r""" r"""
Reshapes a tensor to given target shape; total number of logical elements must Reshapes a tensor to given target shape; total number of logical elements must
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册