会员
周边
众包
新闻
博问
闪存
赞助商
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
yanghh
新随笔
管理
2020年11月28日
Pytorch 神经网络模块之 Linear Layers
摘要: 1. torch.nn.Linear PyTorch 中的 nn.linear() 是用于设置网络中的全连接层的,需要注意的是全连接层的输入与输出都是二维张量,一般形状为 [batch_size, size]。 """ in_features: 指的是输入矩阵的列数,即输入二维张量形状 [batch
阅读全文
posted @ 2020-11-28 21:21 _yanghh
阅读(3346)
评论(0)
推荐(1)
编辑