torch.repeat 张量维度复制

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]])