会员
周边
众包
新闻
博问
闪存
赞助商
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
无左无右
博客园
首页
新随笔
联系
订阅
管理
2024年9月22日
torch.stack
摘要: 看一下stack的直观解释,动词可以简单理解为:把……放成一堆、把……放成一摞。 torch.stack方法用于沿着一个新的维度 join(也可称为cat)一系列的张量(可以是2个张量或者是更多),它会插入一个新的维度,并让张量按照这个新的维度进行张量的cat操作。值得注意的是:张量序列中的张量必须
阅读全文
posted @ 2024-09-22 23:18 无左无右
阅读(57)
评论(0)
推荐(0)
编辑
公告