torch.Tensor.repeat — PyTorch 2.7 documentation (original) (raw)

Tensor.repeat(*repeats) → Tensor¶

Repeats this tensor along the specified dimensions.

Unlike expand(), this function copies the tensor’s data.

Parameters

repeat (torch.Size, int... , tuple of int or list of int) – The number of times to repeat this tensor along each dimension

Example:

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