2024年7月4日
pytorch中的repeat和repeat_interleave
摘要： 个人的简单理解： repeat可以理解为多次复制张量后在指定维度上concate上去，即x.repeat(n,dim=k)等价成torch.cat([x for _ in range(n)],dim=k) repeat_interleave实际上等价于repeat在高一维的基础上运算后再view，即 阅读全文
posted @ 2024-07-04 10:12 kksk43 阅读(156) 评论(0) 推荐(0) 编辑
特效
黑夜
侧边栏隐藏
点击右上角即可分享
微信分享提示