Skip to content

Commit 1bd6c6d

Browse files
Update python/paddle/tensor/manipulation.py
1 parent 89b22cc commit 1bd6c6d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

python/paddle/tensor/manipulation.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6128,9 +6128,9 @@ def repeat_interleave(
61286128
the entries in ``repeats`` which is a int or a Tensor.
61296129
61306130
The image illustrates a typical case of the repeat_interleave operation.
6131-
Given a tensor [[1, 2, 3], [4, 5, 6]], with the repeat counts repeats = [3, 2, 1] and parameter axis = 1, it means that the elements in the 1st column are repeated 3 times, the 2nd column is repeated 2 times, and the 3rd column is repeated 1 time.
6131+
Given a tensor ``[[1, 2, 3], [4, 5, 6]]``, with the repeat counts ``repeats = [3, 2, 1]`` and parameter ``axis = 1``, it means that the elements in the 1st column are repeated 3 times, the 2nd column is repeated 2 times, and the 3rd column is repeated 1 time.
61326132
6133-
The final output is a 2D tensor: [[1, 1, 1, 2, 2, 3], [4, 4, 4, 5, 5, 6]].
6133+
The final output is a 2D tensor: ``[[1, 1, 1, 2, 2, 3], [4, 4, 4, 5, 5, 6]]``.
61346134
61356135
.. image:: https://githubraw.cdn.bcebos.com/PaddlePaddle/docs/develop/docs/images/api_legend/repeat_interleave.png
61366136
:width: 500

0 commit comments

Comments
 (0)