pytorch学习笔记1:nn.Conv2d

我们需要使用torch.nn包中的工具来构建神经网络
首先明确构建一个神经网络需要以下几步:

  1. 定义神经网络的权重,搭建网络结构
  2. 遍历整个数据集进行训练遍历整个数据集进行训练
    将数据输入神经网络
    计算loss
    计算网络权重的梯度
    更新网络权重
    weight = weight + learning_rate * gradient

nn.Conv2d的功能是:对由多个输入平面组成的输入信号进行二维卷积

输入信号的形式为(N,Cin,H,W)
N表示batch size,Cin表示channel个数,H ,W 分别表示特征图的高和宽。
参数说明:
参数

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值