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 分别表示特征图的高和宽。
参数说明:
pytorch学习笔记1:nn.Conv2d_第1张图片

你可能感兴趣的:(学习笔记,python,PyTorch)