码迷,mamicode.com
首页 > 其他好文 > 详细

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

时间:2019-11-14 21:49:22      阅读:778      评论:0      收藏:0      [点我收藏+]

标签:simple   esc   example   data   dimen   code   UNC   monit   extern   

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

标签:simple   esc   example   data   dimen   code   UNC   monit   extern   

原文地址:https://www.cnblogs.com/qinduanyinghua/p/11862492.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!