【学习笔记】【Pytorch】十、线性层
学习地址
主要内容
一、前言
在神经网络中,我们通常用线性层来完成两层神经元间的线性变换。
在普通神经网络里,输入是一个二维矩阵,不需要摊平。而在卷积神经网络里,在网络的最后几层里,会把卷积层摊平放到全连接里进行计算。
注:当把一个输出张量从卷积层传递到线性层时,需要进行 flatten 操作,对一个张量进行flatten(扁平化)。
二、Pytorch的线性层
三、Linear类的使用
from torch.nn import Linear
作用:对输入数据做线性变换:y=Ax+b。
1.使用说明
【实例化】Linear(in_features, out_features, bias=True, device=None, dtype=None)
- 作用:创建一个实例。
- in_features:输入结点数
out_features:输出结点数
bias :是否需要偏置 - 计算公式:
2.代码实现
import torch
import torchvision