torch.Tensor.repeat

Tensor.repeat(*repeats) → Tensor

Повторяет этот тензор вдоль указанных измерений.

В отличие от expand(), эта функция копирует данные тензора.

Предупреждение

repeat() работает иначе, чем numpy.repeat, но больше похож на numpy.tile. См. torch.repeat_interleave(), чтобы использовать оператор, аналогичный numpy.repeat.

Параметры:

repeat (torch.Size, int..., tuple of int or list of int) – Количество повторений этого тензора вдоль каждого измерения

Пример:

>>> x = torch.tensor([1, 2, 3])
>>> x.repeat(4, 2)
tensor([[ 1,  2,  3,  1,  2,  3],
        [ 1,  2,  3,  1,  2,  3],
        [ 1,  2,  3,  1,  2,  3],
        [ 1,  2,  3,  1,  2,  3]])
>>> x.repeat(4, 2, 1).size()
torch.Size([4, 2, 3])