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)