type
Page
status
Invisible
date
Nov 6, 2025
slug
summary
小土堆深度学习快速入门教程学习笔记08
tags
机器学习
深度学习
category
icon
password
Neural Network(NN).
torch.nn.Module 是 PyTorch 中用于构建神经网络的基础类（基类）。
nn.Module 提供的核心功能
功能 | 说明 |
forward() | 定义前向传播 |
parameters() | 返回模型所有需要优化的参数 |
modules() | 返回所有子模块 |
.to(device) | 把模型移动到 GPU/CPU |
.state_dict() | 获取可保存的参数字典 |
.load_state_dict() | 加载参数 |
nn,Module内置了__call__函数，调用时自动加载forward函数