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() - Python技术站