numpy和pytorch的repeat

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

numpy：

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

它是逐个元素地重复。

 

pytorch：

[tensor].repeat_interleave(repeats, dim)

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

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

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