pytorch搭建神经网络过程

2023-05-16

1、torch.nn.Conv2d()

卷积层

用法:

  • Conv2d(in_channels, out_channels, kernel_size, stride=1,padding=0, dilation=1, groups=1,bias=True, padding_mode=‘zeros’)

参数:

in_channels:输入通道数【必填】,常见RGB图像输入通道数为3

out_channels:输出通道数【必填】,具体数值见网络结构

kernel_size:卷积核的大小,类型为int 或者元组,当卷积是正方形的时候,只需要一个整数边长即可,卷积不是正方形,要输入一个元组表示高和宽。【必填】,见具体网络结构

stride:步长,默认为1

padding:填充,设置在所有边界增加值为 0 的边距的大小,例如3*3结构,padding=1,变成5*5结构

dilation:卷积核之间间距,以3*3卷积核为例:

当dilation=0时:

当dilation=1时:

 groups:控制输入和输出之间的连接。

bias: 是否将一个 学习到的 bias 增加输出中,默认是 True 。

padding_mode : 字符串类型,“zeros” 或 “circular”。

 2、torch.nn.MaxPool2d()

最大池化层

用法:torch.nn.MaxPool2d(kernel_size, stride=None, padding=0, dilation=1, return_indices=False, ceil_mode=False)

参数:

kernel_size:最大池化窗口的尺寸

stride:步长,默认值为kernel_size

padding:填充

dilation:窗口中元素关系

return_indices :若设为True,会返回输出最大值的序号,上采样操作会用到

ceil_mode :若设为True,计算输出信号大小的时候,会使用向上取整,代替默认的向下取整的操作

3、torch.nn.ReLU()

线性整流函数

将输入小于0的值幅值为0,输入大于0的值不变。

4、torch.nn.Linear()

全连接层

用法:torch.nn.Linear(in_features,out_features,bias=True)

5、torch.nn.Dropout()

用法:torch.nn.Dropout(p)

p:丢弃多少(0.5-丢弃百分之五十)

本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

pytorch搭建神经网络过程 的相关文章

随机推荐

  • 基于STM32F103系列芯片使用IIC协议来写oled(硬件IIC)

    提示 xff1a 文章写完后 xff0c 目录可以自动生成 xff0c 如何生成可参考右边的帮助文档 文章目录 前言一 协议是什么IIC xff1f 二 相关代码 1 main c2 oled c总结 前言 提示 xff1a 这里可以添加本
  • linux里 read函数用法

    Read 函数 从打开的 fd 设备或文件中读取 count 个字节到 buf 中 ssize tread int fd void buf size tcount fd 文件描述符 buf 读入数据的首地址 count 读入数据的个数 返回
  • linux读写锁

    这里写目录标题 读写锁的认识读写锁的相关函数练习 读写锁的认识 xff08 1 xff09 读写锁是一把锁 xff08 2 xff09 读写锁的类型 xff1a pthread rwlock t lock 又分 读锁 xff08 对内存进行
  • stm32智能家居+微信小程序接收控制

    这里写目录标题 项目介绍mqtt服务器相关知识 在这里插入图片描述 https img blog csdnimg cn 9ad065fb8fac48b1b975fc3a48b99763 png 下位机代码项目需要的一些开发工具 项目介绍 本
  • PWM互补输出,以及死区时间计算

    本文基于野火例程进行解说 实验内容 本次实验输出一对互补的pwm波 xff0c 且进行死区时间的计算说明 代码 互补输出对应的定时器初始化代码 xff1a bsp advance tim c span class token comment
  • 正点原直流有刷驱动板的硬件解说

    这里写目录标题 H桥电流采集电路电源电压采集电路温度采集电路过流检测电路过流保护电路 请添加图片描述 https img blog csdnimg cn fa130e17e23b471e8e27ae942e1ff125 png H桥 因为光
  • stm32103ZET6使用编码器(磁电增量式)

    这里写目录标题 磁电增量式编码器介绍TIM定时器 xff08 编码器接口模式 xff09 一些用到的算法均值滤波冒泡排序 xff08 从小到大 xff09 一阶低通滤波 编码器测数代码编码器接口HAL库函数 正点原子的电机例程 原例程用的是
  • PID算法(位置式pid算法和增量式pid算法)

    这里写目录标题 PID算法介绍比例环节比例 43 积分环节比例 43 积分 43 微分环节 位置式PID增量式PIDPID参数整定采样周期选择PID参数整定方法 请添加图片描述 https img blog csdnimg cn 849bf
  • C++ 关于编译出现“undefined reference to `std::cout‘“的问题

    1 问题概述 在使用gcc编译c 43 43 代码时会出现undefined reference to 96 std cout 39 xff0c 如编译如下代码 xff1a include lt iostream gt using name
  • 在ubuntu20.4下安装ROS-noetic

    换源方法 xff1a 打开software amp updates 在里面找到china 然后选择对应的源 1 添加ROS软件源 将以下命令复制到ubuntu的终端执行 sudo sh c 39 echo 34 deb http packa
  • ubuntu安装pip,setuptools

    https pypi python org pypi pip 解压下载后的文件 xff1a azheng 64 ubuntu16 tar zxvf pip 10 0 0 tar gz 进入目录后安装 xff1a 提示setuptools模块
  • 用putty连接ubuntu

    1 软件更新 sudo apt get update 2 下载openssh server sudo apt get install openssh server 3 查看当前tcp监听端口 netstat tl 4 查看是否安装成功 ps
  • ROS2话题入门

    1 订阅发布模式 一个节点发布数据到某个话题上 xff0c 另外一个节点就可以通过订阅话题拿到数据 除了上述这种一个节点发布 xff0c 一个节点接受的形式外 xff0c ROS2话题通信其实还可以是1对n xff0c n对1 xff0c
  • 比较程序查询方式、程序中断方式、DMA方式对CPU工作效率的影响

    xff08 1 xff09 程序查询方式是由CPU通过程序不断查询I O设备是否已经做好准备 xff0c 从而控制I O设备与主机交换信息 xff0c 只要一启动I O设备 xff0c CPU便不断查询I O设备的准备情况 xff0c 从而
  • ros_canopen使用心得

    近期开始准备上位机与机器人底盘进行CAN通讯 xff0c 花了很长时间在网上整理学习资料 xff0c 现将自己的心得和参考的链接整理如下 首先 xff0c 机器人操作系统中有ros canopen可以去调用 xff0c 其中socketca
  • ROS调用功能包的使用

    在我们学习做项目的时候难免会使用别人的功能包 我在这其中也遇到了许多坑 我在学习motion planning的时候想实现一下tangent bug算法于是就在giuhub找功能包 1是否需要其他依赖包 找到之后就把功能包放在工作空间的sr
  • conda中镜像源添加与删除

    1 查看当前已经添加的镜像源 conda config show sources 2 镜像源添加 conda config add channels http 3 镜像源删除 conda config remove channels htt
  • uwsgi常用指令

    1 uwsgi安装 pip install uwsgi 2 配置uwsgi touch uwsgi ini uwsgi 项目路径 chdir 61 srv xxxx Flask 的uwsgi文件 wsgi file 61 srv xxxx
  • TypeError: conv2d(): argument ‘input‘ (position 1) must be Tensor, not tuple

    搭建网络过程遇到这个问题 xff0c 检查训练过程 xff0c 已经将输入数据通过 transforms ToTensor 转换为Tensor格式 经检查发现是网络结构问题 网络搭建过程中MaxPool2d xff08 xff09 中设置r
  • pytorch搭建神经网络过程

    1 torch nn Conv2d 卷积层 用法 xff1a Conv2d in channels out channels kernel size stride 61 1 padding 61 0 dilation 61 1 groups