megengine.functional.repeat¶

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

重复数组的元素。

参数

inp (Tensor) – 输入张量。
repeats (int) – 每个元素的重复次数。
axis (Optional[int]) – 重复值所沿的轴。默认情况下，使用平展的输入数组，并返回平展的输出数组。

返回

输出张量。

实际案例

import numpy as np
import megengine.functional as F
from megengine import tensor

x = tensor([[1, 2], [3, 4]], np.int32)
y = F.repeat(x, 2, axis=0)
print(y.numpy())

输出：

[[1 2]
 [1 2]
 [3 4]
 [3 4]]

megengine.functional.tile

megengine.functional.roll