torch.Tensor.repeat
-
Tensor.repeat(*sizes) → Tensor -
Повторяет этот тензор вдоль указанных измерений.
В отличие от
expand(), эта функция копирует данные тензора.Предупреждение
repeat()ведет себя иначе, чем numpy.repeat, но более похожа на numpy.tile. Для оператора, аналогичногоnumpy.repeat, см.torch.repeat_interleave().- Параметры:
-
размеры (torch.Size или 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])
© 2024, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://pytorch.org/docs/1.13/generated/torch.Tensor.repeat.html