p60975123
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix docstring, add supported types

This commit is contained in:
Peilin Wang 2020-12-14 14:14:08 -05:00
parent d88aa05859
commit fea11310f0
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mindspore/ops/composite/array_ops.py
View File

@ -59,13 +59,15 @@ def repeat_elements(x, rep, axis=0):
Repeat elements of a tensor along an axis, like np.repeat. Repeat elements of a tensor along an axis, like np.repeat.
Args: Args:
- **x** (Tensor) - The tensor to repeat values for. - **x** (Tensor) - The tensor to repeat values for. Must be of type: float16,
float32, int8, uint8, int16, int32, or int64.
- **rep** (int) - The number of times to repeat, must be positive, required. - **rep** (int) - The number of times to repeat, must be positive, required.
- **axis** (int) - The axis along which to repeat, default 0. - **axis** (int) - The axis along which to repeat, default 0.
Outputs: Outputs:
One tensor with values repeated along the specified axis. If x has shape One tensor with values repeated along the specified axis. If x has shape
(s1, s2, ..., sn) and axis is i, the output will have shape (s1, s2, ..., si * rep, ..., sn) (s1, s2, ..., sn) and axis is i, the output will have shape (s1, s2, ...,
si * rep, ..., sn). The output type will be the same as the type of `x`.
Supported Platforms: Supported Platforms:
``Ascend`` ``GPU`` ``CPU`` ``Ascend`` ``GPU`` ``CPU``