pytorch -- torch.nn.Module

  1. 基础

    torch.nn 是 PyTorch 中用于构建神经网络的模块。nn.Module包含网络各层的定义及forward方法。

    在用户自定义神经网络时,需要继承自nn.Module类。通过继承 nn.Module 类,您可以创建自己的神经网络模型,并定义模型的结构和操作。
    torch.nn 模块中常用的一些类和函数

    nn.Linear: 线性层,用于定义全连接层。

    nn.Conv2d: 二维卷积层,用于处理图像数据。

    nn.ReLU: ReLU 激活函数。

    nn.Sigmoid: Sigmoid 激活函数。

    nn.Dropout: Dropout 层,用于正则化和防止过拟合。

    nn.CrossEntropyLoss: 交叉熵损失函数,通常用于多类别分类问题。

    nn.MSELoss: 均方误差损失函数,通常用于回归问题。

    nn.Sequential: 顺序容器,用于按顺序组合多个层。

    还能使用 PyTorch 提供的优化器(如 torch.optim)和损失函数来训练和优化模型。

  2. 使用

python 复制代码
import torch
from torch.nn import Module
class yaya(Module):

    def __init__(self):
        super().__init__()
    def forward(self,input):
        output = input+1
        return output

tu = yaya()
x = torch.tensor(1.0)
output = tu(x)
print(output)
相关推荐
charley.layabox3 小时前
8月1日ChinaJoy酒会 | 游戏出海高端私享局 | 平台 × 发行 × 投资 × 研发精英畅饮畅聊
人工智能·游戏
DFRobot智位机器人4 小时前
AIOT开发选型:行空板 K10 与 M10 适用场景与选型深度解析
人工智能
想成为风筝6 小时前
从零开始学习深度学习—水果分类之PyQt5App
人工智能·深度学习·计算机视觉·pyqt
F_D_Z6 小时前
MMaDA:多模态大型扩散语言模型
人工智能·语言模型·自然语言处理
江沉晚呤时6 小时前
在 C# 中调用 Python 脚本:实现跨语言功能集成
python·microsoft·c#·.net·.netcore·.net core
大知闲闲哟6 小时前
深度学习G2周:人脸图像生成(DCGAN)
人工智能·深度学习
飞哥数智坊7 小时前
Coze实战第15讲:钱都去哪儿了?Coze+飞书搭建自动记账系统
人工智能·coze
wenzhangli77 小时前
低代码引擎核心技术:OneCode常用动作事件速查手册及注解驱动开发详解
人工智能·低代码·云原生
电脑能手7 小时前
如何远程访问在WSL运行的Jupyter Notebook
ide·python·jupyter
Edward-tan7 小时前
CCPD 车牌数据集提取标注,并转为标准 YOLO 格式
python