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

feat(mge): remove dimshuffle

GitOrigin-RevId: 9e327ee3afd1325bae70f2f3f84cdfd28a5e9ed9
上级 1e2117f6
......@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册