例子:
先创建a:
再创建b:
将b进行repeat,得到c:
b的维度是:[1,2,1,3]
c = b.repeat(1,1,5,1)
的含义是,在0
维度重复1次,在1
维度重复1次,在2
维度重复5次,在3
维度重复1次(重复1次,即不重复);相当于把b的这一维度的值复制1次复制给c。
参考:
- torch.repeat 张量维度复制:https://www.csdn.net/tags/MtTaIgysODAyNDQ3LWJsb2cO0O0O.html
例子:
先创建a:
再创建b:
将b进行repeat,得到c:
b的维度是:[1,2,1,3]
c = b.repeat(1,1,5,1)
的含义是,在0
维度重复1次,在1
维度重复1次,在2
维度重复5次,在3
维度重复1次(重复1次,即不重复);相当于把b的这一维度的值复制1次复制给c。
参考: