会员
周边
众包
新闻
博问
闪存
赞助商
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
Jaxon
版权所有,转载请注明出处
首页
联系
管理
2024年5月26日
nn.Linear输入输出维度的理解
摘要: 目录 最近写代码发现nn.Linear的输入维度其实可以不是一维的输入。 一般来说,nn.Linear的输入一般是(batch_size, input_dimen)。但是官方文档说明了: input:(,input_dimen) output:(,output_dimen) 因此其实*是可以任意维度
阅读全文
posted @ 2024-05-26 11:12 JaxonYe
阅读(36)
评论(0)
推荐(0)
编辑
公告