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

PyTorch 之 强大的 hub 模块和搭建神经网络进行气温预测

翻译: 2.7. 如何利用帮助文档 深入神经网络 pytorch