diff --git a/python_module/megengine/functional/tensor.py b/python_module/megengine/functional/tensor.py index 9bdfc6202c4bb57c0105e39ec534b0be3d281f92..570bd27f74189f885e6303a683bf51948e9803c7 100644 --- a/python_module/megengine/functional/tensor.py +++ b/python_module/megengine/functional/tensor.py @@ -365,8 +365,7 @@ def shapeof(x: Tensor, axis=None): @wrap_io_tensor def dimshuffle(inp: Tensor, pattern: Iterable[int]) -> Tensor: - r - """ + r""" Swap shapes and strides according to given pattern :param inp: Input tensor @@ -408,8 +407,7 @@ def dimshuffle(inp: Tensor, pattern: Iterable[int]) -> Tensor: @wrap_io_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 remain unchanged @@ -447,8 +445,7 @@ def reshape(inp: Tensor, target_shape: Iterable[int]) -> Tensor: @functools.wraps(dimshuffle) def transpose(*args, **kwargs): - r - """See :func:`dimshuffle` + r"""See :func:`dimshuffle` """ return dimshuffle(*args, **kwargs)