深度学习基础网络代码学习(pytorch)

主要是LeNet5:
 pytorch实战1:手把手教你基于pytorch实现LeNet5_pytorch lenet5-CSDN博客
Relu()激活函数:

ReLu(修正线性单元(Rectified Linear Unit)激活函数)

 nn.ReLU(),

 它执行的运算是将输入中小于零的值变为零,保持大于等于零的值不变。

Linear()函数:
  nn.Linear(in_features=400, out_features=120),

执行了一个操作:它创建了一个线性变换,其中输入特征的维度为400,输出特征的维度为120。它可以改变输出大小是因为它将输入的大小为400的特征向量映射到大小为120的输出向量。这个过程通过矩阵乘法和偏置加法来实现,其中权重矩阵的形状是(400, 120),而偏置的形状是(120,)。

torch.flatten(x, 1):

是 PyTorch 中的一个函数调用,用于将输入张量 x 沿着指定维度(维度索引为 1)展平。这意味着函数会将输入张量 x 的指定维度(在这里是维度索引为 1 的维度,通常是列维度)的所有元素展开成一个一维张量。

def forward(self, x): 

定义神经网络的前向传播过程。

在深度学习中,通常没有直接看到调用forward函数的原因是因为forward函数通常在神经网络的定义中被隐式调用。在PyTorch或TensorFlow等深度学习框架中,我们定义了神经网络的结构和参数,并在数据传递过程中,框架会自动调用forward函数来执行前向传播计算。这样的设计隐藏了底层的实现细节,使得用户可以专注于定义网络结构和训练模型,而不必手动调用forward函数。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值