未验证 提交 2fc479c0 编写于 作者: K kuizhiqing 提交者: GitHub

fix repeat doc, test=document_fix (#38238)

上级 69cfb7a2
...@@ -2593,7 +2593,7 @@ def repeat_interleave(x, repeats, axis=None, name=None): ...@@ -2593,7 +2593,7 @@ def repeat_interleave(x, repeats, axis=None, name=None):
Args: Args:
x (Tensor): The input Tensor to be operated. The data of ``x`` can be one of float32, float64, int32, int64. x (Tensor): The input Tensor to be operated. The data of ``x`` can be one of float32, float64, int32, int64.
repeats (Tensor or int): The number of repetitions for each element. repeats is broadcasted to fit the shape of the given axis. repeats (Tensor or int): The number of repetitions for each element. repeats is broadcasted to fit the shape of the given axis.
axis (int, optional): The dimension in which we manipulate. Default: if None, the output tensor is flatten. axis (int, optional): The dimension in which we manipulate. Default: None, the output tensor is flatten.
name(str, optional): The default value is None. Normally there is no name(str, optional): The default value is None. Normally there is no
need for user to set this property. For more information, please need for user to set this property. For more information, please
refer to :ref:`api_guide_Name`. refer to :ref:`api_guide_Name`.
...@@ -2601,6 +2601,11 @@ def repeat_interleave(x, repeats, axis=None, name=None): ...@@ -2601,6 +2601,11 @@ def repeat_interleave(x, repeats, axis=None, name=None):
Returns: Returns:
Tensor: A Tensor with same data type as ``x``. Tensor: A Tensor with same data type as ``x``.
Examples:
.. code-block:: python
import paddle
x = paddle.to_tensor([[1, 2, 3], [4, 5, 6]]) x = paddle.to_tensor([[1, 2, 3], [4, 5, 6]])
repeats = paddle.to_tensor([3, 2, 1], dtype='int32') repeats = paddle.to_tensor([3, 2, 1], dtype='int32')
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册