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

fix(mge): fix tensor docstring

GitOrigin-RevId: 4dab166976f31d78062a044c4cf1ae0a6531b76d
上级 ea93927d
...@@ -365,8 +365,7 @@ def shapeof(x: Tensor, axis=None): ...@@ -365,8 +365,7 @@ def shapeof(x: Tensor, axis=None):
@wrap_io_tensor @wrap_io_tensor
def dimshuffle(inp: Tensor, pattern: Iterable[int]) -> Tensor: def dimshuffle(inp: Tensor, pattern: Iterable[int]) -> Tensor:
r r"""
"""
Swap shapes and strides according to given pattern Swap shapes and strides according to given pattern
:param inp: Input tensor :param inp: Input tensor
...@@ -408,8 +407,7 @@ def dimshuffle(inp: Tensor, pattern: Iterable[int]) -> Tensor: ...@@ -408,8 +407,7 @@ def dimshuffle(inp: Tensor, pattern: Iterable[int]) -> Tensor:
@wrap_io_tensor @wrap_io_tensor
def reshape(inp: Tensor, target_shape: Iterable[int]) -> Tensor: def reshape(inp: Tensor, target_shape: Iterable[int]) -> Tensor:
r r"""
"""
Reshape a tensor to given target shape; total number of logical elements must Reshape a tensor to given target shape; total number of logical elements must
remain unchanged remain unchanged
...@@ -447,8 +445,7 @@ def reshape(inp: Tensor, target_shape: Iterable[int]) -> Tensor: ...@@ -447,8 +445,7 @@ def reshape(inp: Tensor, target_shape: Iterable[int]) -> Tensor:
@functools.wraps(dimshuffle) @functools.wraps(dimshuffle)
def transpose(*args, **kwargs): def transpose(*args, **kwargs):
r r"""See :func:`dimshuffle`
"""See :func:`dimshuffle`
""" """
return dimshuffle(*args, **kwargs) return dimshuffle(*args, **kwargs)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册