会员
周边
新闻
博问
闪存
众包
赞助商
Chat2DB
所有博客
当前博客
我的博客
我的园子
账号设置
会员中心
简洁模式
...
退出登录
注册
登录
krasitz
博客园
首页
新随笔
联系
订阅
管理
05 2022 档案
浅谈Pytorch广播机制
摘要:广播机制 从后往前依次检查维度,如果两个张量对应的维度上数目相等,则会按照该维度相加 若其中一个维度数目为1,则会应用广播机制. 如: a = torch.arange(3).reshape([1, 3, 1]) b = torch.arange(3).reshape([1, 3, 1]) a +
阅读全文
posted @
2022-05-03 20:45
Krasitz
阅读(172)
评论(0)
推荐(0)
公告