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).

Примечание

Эта функция похожа на функцию tile из NumPy.

Параметры

• 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]])