[torch] torch.repeat维度复制
例子：先创建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/MtTaIgysODAyND
·
例子：
先创建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
有“AI”的1024 = 2048，欢迎大家加入2048 AI社区
更多推荐
2
0- 0
已为社区贡献5条内容
所有评论(0)