repeat
(*sizes
)
repeat 相当于一个broadcasting的机制 沿着指定的维度重复tensor。不同与expand()，本函数复制的是tensor中的数据。
import torch
import torch
.nn
.functional
as F
import numpy
as np
a
= torch
.Tensor
(128,1,512)
B
= a
.repeat
(1,5,1)
print(B
.shape
)
torch
.Size
([128, 5, 512])
