07 2024 档案

摘要:个人的简单理解: 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 阅读(152) 评论(0) 推荐(0) 编辑

特效
黑夜
侧边栏隐藏
点击右上角即可分享
微信分享提示