megengine.functional.tile¶

tile(inp, reps)[源代码]¶

通过重复输入给定的次数构造一个数组。如果 reps 的长度为d，则结果的维度为 max(d, inp.ndim)。要求 `` d >= inp.dim``。如果``inp.ndim < d``，则通过添加新轴将输入提升为 d 维。

参数

inp (Tensor) – 输入张量。
reps (Iterable[int]) – 沿每个轴输入的重复次数。

返回

输出张量。

实际案例

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

megengine.functional.split

megengine.functional.repeat