pytorch中的torch.repeat()函数与numpy.tile()

repeat(*sizes) → Tensor

Repeats this tensor along the specified dimensions.

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

WARNING

torch.repeat() behaves differently from numpy.repeat, but is more similar to numpy.tile. For the operator similar to numpy.repeat, see torch.repeat_interleave().

Parameters

sizes (torch.Size or 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])

 

你可能感兴趣的:(pytorch中的torch.repeat()函数与numpy.tile())