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)
相关推荐
IT_陈寒2 小时前
React 18实战:7个被低估的Hooks技巧让你的开发效率提升50%
前端·人工智能·后端
数据智能老司机4 小时前
精通 Python 设计模式——分布式系统模式
python·设计模式·架构
逛逛GitHub4 小时前
飞书多维表“独立”了!功能强大的超出想象。
人工智能·github·产品
机器之心4 小时前
刚刚,DeepSeek-R1论文登上Nature封面,通讯作者梁文锋
人工智能·openai
数据智能老司机5 小时前
精通 Python 设计模式——并发与异步模式
python·设计模式·编程语言
数据智能老司机5 小时前
精通 Python 设计模式——测试模式
python·设计模式·架构
数据智能老司机5 小时前
精通 Python 设计模式——性能模式
python·设计模式·架构
c8i5 小时前
drf初步梳理
python·django
每日AI新事件5 小时前
python的异步函数
python
这里有鱼汤6 小时前
miniQMT下载历史行情数据太慢怎么办?一招提速10倍!
前端·python