mindspore.ops.repeat_interleave

mindspore.ops.repeat_interleave(x, repeats, dim=None)[源代码]

沿着轴重复Tensor的元素，类似 numpy.Repeat。

参数：

x (Tensor) - 进行重复操作的入参Tensor，类型必须为float16，float32，int8，uint8，int16，int32或者int64。
repeats (int) - 指定复制次数，为正数。
dim (int，可选) - 指定复制轴，如果为None，则默认为0。

返回：

Tensor，值沿指定轴复制。如果输入的shape为 \((s1, s2, ..., sn)\) ，轴为i，则输出的shape为 \((s1, s2, ..., si * repeats, ..., sn)\) 。输出的数据类型与输入相同。

支持平台：

Ascend GPU CPU

样例：

>>> x = Tensor(np.array([[0, 1, 2], [3, 4, 5]]), mindspore.int32)
>>> output = ops.repeat_interleave(x, repeats=2, dim=0)
>>> print(output)
[[0 1 2]
 [0 1 2]
 [3 4 5]
 [3 4 5]]