pytorch中的神经网络子模块(线性模块)——torch.nn.Linear
Posted
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了pytorch中的神经网络子模块(线性模块)——torch.nn.Linear相关的知识,希望对你有一定的参考价值。
参考技术A pytorch中的线性模块的实现如下,在init函数中定义weight值和bias值。所以若要对linear子模块的参数进行初始化,利用如下策略可以对单个linear子模块进行参数初始化。
以上是关于pytorch中的神经网络子模块(线性模块)——torch.nn.Linear的主要内容,如果未能解决你的问题,请参考以下文章
Pytorch Note37 PyTorch 中的循环神经网络模块
pytorch中的顺序容器——torch.nn.Sequential
PyTorch从入门到精通100讲-神经网络在pytorch中的应用
PyTorch-torch04: Cov2d二维卷积和torch.nn