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)
相关推荐
时序之心1 分钟前
时空数据挖掘五大革新方向详解篇!
人工智能·数据挖掘·论文·时间序列
IMPYLH28 分钟前
Python 的内置函数 reversed
笔记·python
.30-06Springfield32 分钟前
人工智能概念之七:集成学习思想(Bagging、Boosting、Stacking)
人工智能·算法·机器学习·集成学习
说私域2 小时前
基于开源AI智能名片链动2+1模式S2B2C商城小程序的超级文化符号构建路径研究
人工智能·小程序·开源
永洪科技2 小时前
永洪科技荣获商业智能品牌影响力奖,全力打造”AI+决策”引擎
大数据·人工智能·科技·数据分析·数据可视化·bi
shangyingying_12 小时前
关于小波降噪、小波增强、小波去雾的原理区分
人工智能·深度学习·计算机视觉
小赖同学啊2 小时前
物联网数据安全区块链服务
开发语言·python·区块链
码荼3 小时前
学习开发之hashmap
java·python·学习·哈希算法·个人开发·小白学开发·不花钱不花时间crud
书玮嘎3 小时前
【WIP】【VLA&VLM——InternVL系列】
人工智能·深度学习
猫头虎3 小时前
猫头虎 AI工具分享:一个网页抓取、结构化数据提取、网页爬取、浏览器自动化操作工具:Hyperbrowser MCP
运维·人工智能·gpt·开源·自动化·文心一言·ai编程