mindspore.Tensor.repeat

查看源文件
mindspore.Tensor.repeat(repeats, axis=None)[源代码]

对数组中的元素进行重复复制。

参数:
  • repeats (Union[int, tuple, list]) - 每个元素的重复次数,repeats 被广播以适应指定轴的shape。

  • axis (int, 可选) - 轴方向上的重复值。默认情况下,使用展开的输入Tensor,并返回一个展开的输出Tensor。默认值: None

返回:

Tensor,除了维度外,与输入Tensor具有相同的shape。

异常:
  • ValueError - 维度超出范围。

  • TypeError - 参数类型不匹配。

支持平台:

Ascend GPU CPU

样例:

>>> import numpy as np
>>> from mindspore import Tensor
>>> x = Tensor(np.array(3))
>>> print(x.repeat(4))
[3 3 3 3]
>>> x = Tensor(np.array([[1, 2],[3, 4]]))
>>> print(x.repeat(2))
[1 1 2 2 3 3 4 4]
>>> print(x.repeat(3, axis=1))
[[1 1 1 2 2 2]
[3 3 3 4 4 4]]
>>> print(x.repeat([1,2], axis=0))
[[1 2]
[3 4]
[3 4]]