PyTorch中的torch.nn.Linear函数解析-Toy模板网

这篇具有很好参考价值的文章主要介绍了PyTorch中的torch.nn.Linear函数解析。希望对大家有所帮助。如果存在错误或未考虑完全的地方，请大家不吝赐教，您也可以点击"举报违法"按钮提交疑问。

torch.nn是包含了构筑神经网络结构基本元素的包，在这个包中，可以找到任意的神经网络层。这些神经网络层都是nn.Module这个大类的子类。torch.nn.Linear就是神经网络中的线性层，可以实现形如y=Xweight^T+b的加和功能。

nn.Linear()：用于设置网络中的全连接层，需要注意的是全连接层的输入与输出都是二维张量
一般形状为[batch_size, size]，不同于卷积层要求输入输出是四维张量。

其用法与形参说明如下：
PyTorch中的torch.nn.Linear函数解析,pytorch,人工智能,python

代码很简单，但有许多细节需要声明：

1）nn.Linear是一个类，使用时进行类的实例化
2）实例化的时候，nn.Linear需要输入两个参数，in_features为上一层神经元的个数，out_features为这一层的神经元个数
3）不需要定义w和b。所有nn.Module的子类，形如nn.XXX的层，都会在实例化的同时随机生成w和b的初始值。所以实例化之后，我们就可以调用属性weight和bias来查看生成的w和b。其中w是必然会生成的，b是我们可以控制是否要生成的。在nn.Linear类中，有参数bias，默认 bias = True。如果我们希望不拟合常量b，在实例化时将参数bias设置为False即可。
4）由于w和b是随机生成的，所以同样的代码多次运行后的结果是不一致的。如果我们希望控制随机性，则可以使用torch中的random类。如：torch.random.manual_seed(420) #人为设置随机数种子
5）由于不需要定义常量b，因此在特征张量中，不需要留出与常数项相乘的那一列，只需要输入特征张量。
6）输入层只有一层，并且输入层的结构（神经元的个数）由输入的特征张量X决定，因此在PyTorch中构筑神经网络时，不需要定义输入层。
7）实例化之后，将特征张量输入到实例化后的类中。
参考链接：https://huaweicloud.csdn.net/6380304edacf622b8df867e7.html?spm=1001.2101.3001.6650.1&utm_medium=distribute.pc_relevant.none-task-blog-2_defaultCTRLIST_{activity-1-122750531-blog-127900126.235%5Ev38%5Epc_relevant_anti_t3&depth_1-utm_source=distribute.pc_relevant.none-task-blog-2}default_CTRLISTactivity-1-122750531-blog-127900126.235%5Ev38%5Epc_relevant_anti_t3&utm_relevant_index=2文章来源地址https://www.toymoban.com/news/detail-571634.html

到了这里，关于PyTorch中的torch.nn.Linear函数解析的文章就介绍完了。如果您还想了解更多内容，请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章，希望大家以后多多支持TOY模板网！