在项目中发现这两个同名的函数的效果不一样,下面简单介绍:
numpy的repeat
cpp
a = np.array([1,2,3])
print('a', a) ## 结果为123
b = a.repeat(3)
print('b') ## 结果为111222333
pytorch的repeat和repeat_interleave
cpp
labels = np.array([1,2,3])
labels = torch.tensor(labels, dtype=torch.float32)
labels1 = labels.repeat(3)
print(labels1) ##结果为123123123
labels2 = labels.repeat_interleave(3, dim=0)
print(labels2 ) ##结果为111222333
numpy没有repeat_interleave函数。