会员
周边
众包
新闻
博问
闪存
赞助商
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
猪猪猪猪侠
博客园
首页
新随笔
联系
订阅
管理
2022年9月18日
pytorch如何批量reshape,如何每batch_size进行reshape
摘要: 假设我有一个tensor,它的batch_size是2: tensor = torch.randn([2, 6]) print(tensor.shape) 输出是 torch.Size([2, 6]) 其中tensor.shape[0]代表tensor的batch_size 如果我要把其中每个Bat
阅读全文
posted @ 2022-09-18 10:11 猪猪猪猪侠
阅读(368)
评论(0)
推荐(0)
编辑
公告