Megvii Engine Team
5 years ago
1 changed files with 3 additions and 6 deletions
Unified View
Diff Options
+ 3
- 6
python_module/megengine/functional/tensor.py
View File
| @@ -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: | |||||
| @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: | |||||
| @functools.wraps(dimshuffle) | @functools.wraps(dimshuffle) | ||||
| def transpose(*args, **kwargs): | def transpose(*args, **kwargs): | ||||
| r | |||||
| """See :func:`dimshuffle` | |||||
| r"""See :func:`dimshuffle` | |||||
| """ | """ | ||||
| return dimshuffle(*args, **kwargs) | return dimshuffle(*args, **kwargs) | ||||