首页 > 其他分享 >线性层

线性层

时间:2023-08-19 12:00:12浏览次数:44  
标签:self torch imgs import 线性 output

线性层

线性层结构

线性连接是全连接的一种形式,但全连接不一定是线性连接。全连接层可以使用非线性激活函数,而线性连接只进行简单的线性映射。线性连接如下图:

神经网络中的使用

torch.nn.Linear(in_features, out_features, bias=True, device=None, dtype=None)
  • in_features (in) – size of each input sample
  • out_features (int) – size of each output sample
  • bias (bool) – If set to False, the layer will not learn an additive bias. Default: True

代码实现

import torch
import torchvision.datasets
from torch import nn
from torch.nn import Linear
from torch.utils.data import DataLoader

# 使用CIFAR10数据集
dataset = torchvision.datasets.CIFAR10("./dataset2", train=False, transform=torchvision.transforms.ToTensor())

dataloader = DataLoader(dataset, batch_size=64, drop_last=True)

class Baselinear(nn.Module):
    def __init__(self):
        super(Baselinear, self).__init__()
        self.linear1 = Linear(196608, 10)

    def forward(self, input):
        output = self.linear1(input)
        return output

baselinear = Baselinear()

for data in dataloader:
    imgs, targets = data
    print(imgs.shape)
    # output = torch.reshape(imgs, (1, 1, 1, -1))
    # flatten是将imgs图片进行展平,与上述的reshape作用相同
    output = torch.flatten(imgs)
    print(output.shape)
    output = baselinear(output)
    print(output.shape)

使用reshape的输出结果:

![image-20230808122914489](C:\User

使用flatten的输出结果:

标签:self,torch,imgs,import,线性,output
From: https://www.cnblogs.com/yq-ydky/p/17632449.html

相关文章

  • 学不会的线性基
    前言最后一次“杭电杯”结束了捏,看到同级的另一个队哐哐过题,感觉自己好菜捏......
  • ITK 实例3 PNG图像进行二维非线性映射
    1#include"itkImage.h"2#include"itkImageFileReader.h"3#include"itkImageFileWriter.h"4//非线性映射滤波器头文件5#include"itkSigmoidImageFilter.h"67intmain(intargc,char*argv[])8{9/*if(argc......
  • ITK 实例4 MHA格式文件进行三维非线性映射
    1#include"itkImage.h"2#include"itkImageFileReader.h"3#include"itkImageFileWriter.h"4//非线性映射滤波器头文件5#include"itkSigmoidImageFilter.h"6//注:非线性映射算法只能实现像素值(0-255)范围的输入输出映射。7intmain(intargc,cha......
  • 【线性代数】线性方程组 如何求方程组的解/基础解系/通解
    1.如何求齐次方程组的基础解系前面已经学过:基础解系的定义为:一个向量组中所有的向量都是原方程的解,并且线性无关,又能由这个向量组线性表出这个方程组的所有解。先讲齐次方程组是因为它右侧常数都为0,解起来更为简单。步骤:先对齐次方程组的系数矩阵作初等行变换,直到化为行阶梯矩......
  • Python用PyMC贝叶斯GLM广义线性模型、NUTS采样器拟合、后验分布可视化
    全文链接:https://tecdat.cn/?p=33436原文出处:拓端数据部落公众号尽管贝叶斯方法相对于频率主义方法的理论优势已经在其他地方进行了详细讨论,但其更广泛采用的主要障碍是“可用性”。而使用贝叶斯方法,客户可以按照自己认为合适的方式定义模型。线性回归在此示例中,我们将帮助客......
  • [学习笔记] 线性基
    你说我一个连线性基都不会的人怎么可能走的远,我跟你说我也是这么想的,但是你先别急。一、线性基OI中常用全部的就是\(2\)进制下的异或线性基。线性基就是可以把一个集合里的数转化成一组基,使得这组基里所有xor出来的结果于原集合xor出来的结果完全一致。这是一个线性基......
  • 线性泛基因组和图形泛基因组
     001、线性泛基因组仅仅包含序列信息。迭代组装、map-to-pan、个体denovo组装的泛基因组均是线性泛基因组。 002、图形泛基因组图结构泛基因组是一个二维序列图谱,它以参考基因组为框架,以单个碱基作为图的节点,碱基间的前后关系作为图的边,存在序列差异的地方会自然形成不同......
  • 线性表【数据结构学习-青岛大学王卓老师】
    https://www.bilibili.com/video/BV1qz4y1p767/线性表线性表的初始化(顺序表)StatusInitList(SqList&L){L.elem=(ElemType*)malloc(sizeof(ElemType)*MAXSIZE);if(!L.elem)exit(OVERFLOW);L.length=0;returnOK;}线性表的销毁voi......
  • 线性规划之单纯形算法
    学了很长时间,一直不是很能理解,所以就准备写一篇。这篇文章只讲单纯形算法。假设我们已经得到了标准型:\[\begin{aligned}\max:\sum\limits_{i=1}^na_ix_i\\\sum\limits_{i=1}^nb_{j,i}x_i=c_j&,j=1,2\dotsm\\x_i\geqslant0&,i=1,2\dotsn\end{aligned}\]而得到最优解的过......
  • 线性筛与欧拉函数
    很萌很可爱!就是把纸质笔记上letex写在这里有亿点慢线性筛埃氏筛,\(O(n\log\logn)\),思路是⌈标记所有质数的倍数⌋这样每个合数可能会被标记好几次,我们改进一下,让每个合数只有一种被标记的方式,即⌈最小质因子*常数⌋具体而言,⌈枚举\(2\tox\)把当前数\(i\)跟......