megengine.functional.repeat

repeat(inp, repeats, axis=None)[源代码]

重复数组的元素。

参数
  • inp (Tensor) – 输入张量。

  • repeats (int) – 每个元素的重复次数。

  • axis (Optional[int]) – 重复值所沿的轴。 默认情况下,使用平展的输入数组,并返回平展的输出数组。

返回

输出张量。

实际案例

>>> import numpy as np
>>> x = Tensor([[1, 2], [3, 4]], np.int32)
>>> F.repeat(x, 2, axis=0)
Tensor([[1 2]
 [1 2]
 [3 4]
 [3 4]], dtype=int32, device=xpux:0)