pytorch中的torch.repeat()函数与numpy.tile()
Posted qinduanyinghua
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了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()的主要内容，如果未能解决你的问题，请参考以下文章