pytorch学习笔记（十二）————全连接层

最新推荐文章于 2025-04-15 12:20:00 发布

南风渐起

最新推荐文章于 2025-04-15 12:20:00 发布

阅读量8.4k

点赞数 1

分类专栏： pytorch

本文链接：https://blog.youkuaiyun.com/weixin_43886278/article/details/104282018

版权

pytorch学习笔记（十二）————全连接层

nn.Linear()
在这里插入图片描述
class torch.nn.Linear（in_features，out_features，bias = True ）

对传入数据应用线性变换：y = A x+ b

参数：

in_features - 每个输入样本的大小

out_features - 每个输出样本的大小

bias - 如果设置为False，则图层不会学习附加偏差。默认值：True

实现代码：

import  torch
import  torch.nn as nn
import  torch.nn.functional as F
import  torch.optim as optim
from    torchvision import datasets, transforms


batch_size=200
learning_rate=0.01
epochs=10

train_loader = torch.utils.data.DataLoader(
    datasets.MNIST('../data', train=True, download=True,
                   transform=transforms.Compose([
                       transforms.ToTensor(),
                       transforms.Normalize((0.1307,), (0.3081,))
                   ])),
    batch_size=batch_size, shuffle