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