【pytorch】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(2,0)
#[[0 1 2]
#[0 1 2]
#[3 4 5]
#[3 4 5]]
c = a.repeat(2,1)
#[[0 0 1 1 2 2]
#[3 3 4 4 5 5]]

torch repeat

参数代表了各轴分别重复几次

import torch
a = torch.arange(6).reshape((2,3))
b = a.repeat(1,2)
# tensor([[0, 1, 2, 0, 1, 2],
#        [3, 4, 5, 3, 4, 5]])
c = a.repeat(2,1)
#tensor([[0, 1, 2],
#        [3, 4, 5],
#        [0, 1, 2],
#        [3, 4, 5]])