未验证 提交 34a256c9 编写于 作者: N Nyakku Shigure 提交者: GitHub

fix paddle.tile en docs, test=document_fix (#40731)

上级 c54c60de
...@@ -1732,12 +1732,12 @@ def tile(x, repeat_times, name=None): ...@@ -1732,12 +1732,12 @@ def tile(x, repeat_times, name=None):
Args: Args:
x (Tensor): The input tensor, its data type should be bool, float32, float64, int32 or int64. x (Tensor): The input tensor, its data type should be bool, float32, float64, int32 or int64.
repeat_times (Tensor|tuple|list): The number of repeating times. If repeat_times is a list or tuple, all its elements repeat_times (list|tuple|Tensor): The number of repeating times. If repeat_times is a list or tuple, all its elements
should be integers or 1-D Tensors with the data type int32. If repeat_times is a Tensor, it should be an 1-D Tensor with the data type int32. should be integers or 1-D Tensors with the data type int32. If repeat_times is a Tensor, it should be an 1-D Tensor with the data type int32.
name (str, optional): Name for the operation (optional, default is None). For more information, please refer to :ref:`api_guide_Name`. name (str, optional): Name for the operation (optional, default is None). For more information, please refer to :ref:`api_guide_Name`.
Returns: Returns:
N-D Tensor. The data type is the same as ``x``. N-D Tensor. The data type is the same as ``x``. The size of the i-th dimension is equal to ``x[i] * repeat_times[i]``.
Examples: Examples:
.. code-block:: python .. code-block:: python
...@@ -1747,16 +1747,18 @@ def tile(x, repeat_times, name=None): ...@@ -1747,16 +1747,18 @@ def tile(x, repeat_times, name=None):
data = paddle.to_tensor([1, 2, 3], dtype='int32') data = paddle.to_tensor([1, 2, 3], dtype='int32')
out = paddle.tile(data, repeat_times=[2, 1]) out = paddle.tile(data, repeat_times=[2, 1])
np_out = out.numpy() np_out = out.numpy()
# [[1, 2, 3], [1, 2, 3]] # [[1, 2, 3]
# [1, 2, 3]]
out = paddle.tile(data, repeat_times=[2, 2]) out = paddle.tile(data, repeat_times=(2, 2))
np_out = out.numpy() np_out = out.numpy()
# [[1, 2, 3, 1, 2, 3], [1, 2, 3, 1, 2, 3]] # [[1, 2, 3, 1, 2, 3]
# [1, 2, 3, 1, 2, 3]]
repeat_times = paddle.to_tensor([2, 1], dtype='int32') repeat_times = paddle.to_tensor([1, 2], dtype='int32')
out = paddle.tile(data, repeat_times=repeat_times) out = paddle.tile(data, repeat_times=repeat_times)
np_out = out.numpy() np_out = out.numpy()
# [[1, 2, 3], [1, 2, 3]] # [[1, 2, 3, 1, 2, 3]]
""" """
if paddle.in_dynamic_mode(): if paddle.in_dynamic_mode():
return _C_ops.tile(x, 'repeat_times', repeat_times) return _C_ops.tile(x, 'repeat_times', repeat_times)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册