torch.repeat 用于张量的制定维度复制
import torch
a = torch.rand(2,3)
# 如果 a 是 feature vector, 则 batch=2(dim=0), channels=3(dim=1)
b = a.repeat(2,1)
# 在 batch(dim=0) 维度上复制 2 次,在 channel(dim=1) 维度上复制 1 次
# 即 repeat 的参数为其维度的复制次数,第 1 个参数表示对第 0 dim 的重复次数,依次类推
######## 结果展示 #########
a = torch.rand(2,3)
a
tensor([[0.4040, 0.7043, 0.0454],
[0.9065, 0.1127, 0.2905]])
a.repeat(2, 1)
tensor([[0.4040, 0.7043, 0.0454],
[0.9065, 0.1127, 0.2905],
[0.4040, 0.7043, 0.0454],
[0.9065, 0.1127, 0.2905]])