【学习笔记】【Pytorch】十、线性层

本文介绍了PyTorch中的线性层Linear,它用于神经网络中的线性变换。在实例化Linear时需指定输入和输出特征数,以及是否包含偏置。文章通过创建一个模型并应用到CIFAR10数据集上,展示了如何将卷积层的输出通过Flatten或reshape转换为适合线性层的形状。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

学习地址

PyTorch深度学习快速入门教程【小土堆】.

主要内容

一、前言

在神经网络中,我们通常用线性层来完成两层神经元间的线性变换。
在这里插入图片描述
在普通神经网络里,输入是一个二维矩阵,不需要摊平。而在卷积神经网络里,在网络的最后几层里,会把卷积层摊平放到全连接里进行计算。
:当把一个输出张量从卷积层传递到线性层时,需要进行 flatten 操作,对一个张量进行flatten(扁平化)。

二、Pytorch的线性层

linear-layers

三、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
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值