【学习笔记】神经网络—线性层
创始人
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

 

相关内容

热门资讯

一分钟内幕!科乐吉林麻将系统发... 一分钟内幕!科乐吉林麻将系统发牌规律,福建大玩家确实真的是有挂,技巧教程(有挂ai代打);所有人都在...
一分钟揭秘!微扑克辅助软件(透... 一分钟揭秘!微扑克辅助软件(透视辅助)确实是有挂(2024已更新)(哔哩哔哩);1、用户打开应用后不...
五分钟发现!广东雀神麻雀怎么赢... 五分钟发现!广东雀神麻雀怎么赢,朋朋棋牌都是是真的有挂,高科技教程(有挂方法)1、广东雀神麻雀怎么赢...
每日必看!人皇大厅吗(透明挂)... 每日必看!人皇大厅吗(透明挂)好像存在有挂(2026已更新)(哔哩哔哩);人皇大厅吗辅助器中分为三种...
重大科普!新华棋牌有挂吗(透视... 重大科普!新华棋牌有挂吗(透视)一直是有挂(2021已更新)(哔哩哔哩)1、完成新华棋牌有挂吗的残局...
二分钟内幕!微信小程序途游辅助... 二分钟内幕!微信小程序途游辅助器,掌中乐游戏中心其实存在有挂,微扑克教程(有挂规律)二分钟内幕!微信...
科技揭秘!jj斗地主系统控牌吗... 科技揭秘!jj斗地主系统控牌吗(透视)本来真的是有挂(2025已更新)(哔哩哔哩)1、科技揭秘!jj...
1分钟普及!哈灵麻将攻略小,微... 1分钟普及!哈灵麻将攻略小,微信小程序十三张好像存在有挂,规律教程(有挂技巧)哈灵麻将攻略小是一种具...
9分钟教程!科乐麻将有挂吗,传... 9分钟教程!科乐麻将有挂吗,传送屋高防版辅助(总是存在有挂)1、完成传送屋高防版辅助透视辅助安装,帮...
每日必看教程!兴动游戏辅助器下... 每日必看教程!兴动游戏辅助器下载(辅助)真是真的有挂(2025已更新)(哔哩哔哩)1、打开软件启动之...