torch.tile

torch.tile(input, dims) → Tensor

Создаёт тензор, повторяя элементы input. Аргумент dims задаёт число повторений в каждом измерении.

Если dims задаёт меньше измерений, чем input, то к dims добавляются единицы в начале, пока не будет задано всех измерений. Например, если input имеет форму (8, 6, 4, 2), а dims — (2, 2), то dims рассматривается как (1, 1, 2, 2).

Аналогично, если у input меньше измерений, чем задано в dims, то input рассматривается как если бы оно было расширено в нулевом измерении до тех пор, пока не будет содержать столько же измерений, сколько указано в dims. Например, если input имеет форму (4, 2), а dims — (3, 3, 2, 2), то input рассматривается как имеющее форму (1, 1, 4, 2).

Примечание

Эта функция аналогична функции NumPy’s tile.

Параметры:

• input (Tensor) – тензор, элементы которого нужно повторить.
• dims (tuple) – число повторений по каждому измерению.

Пример:

>>> x = torch.tensor([1, 2, 3])
>>> x.tile((2,))
tensor([1, 2, 3, 1, 2, 3])
>>> y = torch.tensor([[1, 2], [3, 4]])
>>> torch.tile(y, (2, 2))
tensor([[1, 2, 1, 2],
        [3, 4, 3, 4],
        [1, 2, 1, 2],
        [3, 4, 3, 4]])