【学习笔记】神经网络—线性层
创始人
2024-11-28 17:05:57
0

查看官方文档:PyTorchicon-default.png?t=N7T8https://pytorch.org/

找到Linear Layers,如下所示 

 线性层函数介绍:

CLASStorch.nn.Linear(in_featuresout_featuresbias=Truedevice=Nonedtype=None)[SOURCE]icon-default.png?t=N7T8https://pytorch.org/docs/stable/_modules/torch/nn/modules/linear.html#Linear

参数介绍:

Parameters

  • in_features (int) – size of each input sample

  • out_features (int) – size of each output sample

  • bias (bool) – If set to , the layer will not learn an additive bias. Default: FalseTru

 线性层又叫全连接层,其中每个神经元与上一层所有神经元相连,一个简单的线性层如下图所示:

in_features指的是上图中x1,x2...xd的个数,即d

out_features 指的是上图中g1,g2...gL的个数,即L

bias:偏置,上面加不加b,就由这个bias控制,True时,加b        ·

Variables

  • weight (torch.Tensor) :相当于上图中的

  • bias : 相当于上图中的b

下面以一个简单的网络结果VGG16模型为例 :

 我们想实现下面这个:

import torch import torchvision from torch import nn from torch.nn import Linear from torch.utils.data import DataLoader  dataset=torchvision.datasets.CIFAR10('datasets',train=False,transform=torchvision.transforms.ToTensor(),download=True)  dataloader=DataLoader(dataset,batch_size=64,drop_last=True)  class Tudui(nn.Module):         def __init__(self):             super(Tudui,self).__init__()             self.linear1 = Linear(in_features=196608,out_features=10)         def forward(self,input):             output = self.linear1(input)             return output   tudui=Tudui() for data in dataloader:     imgs,targets=data     print(imgs.shape)   # torch.Size([64, 3, 32, 32])   imgs个数为157,最后imgs[157]个数为14张图片     # output=torch.reshape(imgs,([1,1,1,-1]))  #最后一个-1代表让其自己计算     output = torch.flatten(imgs)     print(output.shape)  #torch.    Size([1, 1, 1, 196608])     output= tudui(output)     print(output.shape)  #经过神经网络 torch.Size([1, 1, 1, 10])

如下图所示,pytorch提供的一些网络模型,像AlexNet,VGG,ResNet等

 语义分割:Semantic Segmentationicon-default.png?t=N7T8https://pytorch.org/vision/0.9/models.html#semantic-segmentation

目标检测、实例分割和人体结构检测(行为检测) 

  • Object Detection, Instance Segmentation and Person Keypoint Detectionicon-default.png?t=N7T8https://pytorch.org/vision/0.9/models.html#object-detection-instance-segmentation-and-person-keypoint-detection

 

相关内容

热门资讯

一分钟了解!雀神辅助器最新款(... 一分钟了解!雀神辅助器最新款(辅助)切实有挂(2022已更新)(哔哩哔哩);1、该软件可以轻松地帮助...
九分钟辅助挂!天天摸摸麻将有挂... 九分钟辅助挂!天天摸摸麻将有挂吗“详细透视辅助挂教程”原来真的有挂1、每一步都需要思考,不同水平的挑...
必备攻略!全民牛牛中心辅助器(... 必备攻略!全民牛牛中心辅助器(辅助挂)外挂透明挂辅助挂(2020已更新)(哔哩哔哩)全民牛牛中心辅助...
八分钟发现!小闲巴渝棋牌有挂吗... 八分钟发现!小闲巴渝棋牌有挂吗,aapoKer都是存在有挂,大神讲解(有挂透视)1、用户打开应用后不...
玩家必看秘籍!边锋老友二打一辅... 玩家必看秘籍!边锋老友二打一辅助(辅助)竟然真的是有挂(2025已更新)(哔哩哔哩);1、下载好边锋...
13钟辅助挂!白金岛跑得快“详... 13钟辅助挂!白金岛跑得快“详细透视辅助神器教程”原来真的有挂1、每一步都需要思考,不同水平的挑战会...
重大科普!聚友互娱手机app辅... 您好,聚友互娱手机app辅助器这款游戏可以开挂的,确实是有挂的,需要了解加微【757446909】很...
五分钟科普!哥哥跑得快有挂吗,... 五分钟科普!哥哥跑得快有挂吗,epoker好像真的有挂,安装教程(有挂揭秘)1、哥哥跑得快有挂吗系统...
盘点十款!六瓣数字消有没有挂(... 盘点十款!六瓣数字消有没有挂(透明挂)确实有挂(2025已更新)(哔哩哔哩);1、玩家可以在六瓣数字...
1分钟辅助挂!微友麻将助手神器... 1分钟辅助挂!微友麻将助手神器“详细透视辅助脚本教程”原来真的有挂是一款可以让一直输的玩家,快速成为...