【AI】PyTorch入门（五）：构建神经模型

郭老二

已于 2023-05-10 21:57:53 修改

阅读量498

点赞数 1

分类专栏： AI 文章标签： pytorch 人工智能深度学习

于 2022-09-07 19:39:52 首次发布

本文链接：https://blog.csdn.net/u010168781/article/details/126751775

版权

AI 专栏收录该内容

106 篇文章 69 订阅 ¥69.90 ¥99.00

订阅专栏

本文介绍了PyTorch中构建神经网络的基础，包括nn.Flatten的展开操作、nn.Linear的线性变化、nn.ReLU激活函数、nn.Sequential顺序容器、nn.Softmax的使用，以及参数模型的概览。

摘要由CSDN通过智能技术生成

AI学习目录汇总

1、简述

torch.nn命名空间提供了构建自己的神经网络所需的所有构建模块，PyTorch中的每个模块都是 nn.Module 的子类。
神经网络由各种模块组成，它本身也是模块，也就是说神经网络是一个嵌套结构，这种嵌套结构方便构建和管理复杂的架构。

2、网络层详解

创建一个简单网络，并打印…

import torch
from torch import nn

device = "cuda" if torch.cuda.is_available() else "cpu"

class NeuralNetwork(nn.Module):
    def __init__(self):
        super(NeuralNetwork, self).__init__()
        self.flatten = nn.Flatten()
        self.linear_relu_stack = nn.Sequential(
            nn.Linear(28*28, 512),
            nn.ReLU(),

了解本专栏