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 - 参数类型不匹配。
- 其他API参考:
mindspore.Tensor.reshape()
:为Tensor指定新的shape,不更改Tensor的数据。mindspore.Tensor.resize()
:就地改变Tensor的shape和size。
- 支持平台:
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]]