会员
周边
新闻
博问
闪存
众包
赞助商
Chat2DB
所有博客
当前博客
我的博客
我的园子
账号设置
会员中心
简洁模式
...
退出登录
注册
登录
kksk43
博客园
首页
新随笔
联系
订阅
管理
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
阅读(390)
评论(0)
推荐(0)
公告
特效
黑夜
侧边栏隐藏