numpy和pytorch的repeat

因此曾经踩过坑,吃了大亏,所以这次要记录一下。

numpy:

np.repeat()  <==>  [array].repeat(repeats, axis)

它是逐个元素地重复。

numpy和pytorch的repeat_第1张图片

 

pytorch:

[tensor].repeat_interleave(repeats, dim)

它才是等价于上面np.repeat  !!!

[tensor].repeat(第一维复制的次数,第二维复制的次数,....)

它是将这一维度整个tensor地重复。

numpy和pytorch的repeat_第2张图片

 

 

你可能感兴趣的:(pytorch,numpy,python)