【pytorch】numpy repeat 和 torch repeat numpy repeat torch repeat numpy repeat numpy.repeat(a,repeats,axis=None) axis=0,沿着y轴复制,增加行数 axis=1,沿着x轴复制,增加列数 axis=None,变成一个行向量 import numpy as np a = np.arange(6).reshape((2,3)) b = a.repeat<