株洲防火门胶 构建经网络 — PyTorch Tutorials 2.1.+cu128 文档

保温护角专用胶 构建经网络#

创建于:221年2月9日 | 后新:225年1月24日 | 后验证:未验证株洲防火门胶

经网络由执行数据操作的层/模块组成。torch.nn 命名空间提供了构建自定义经网络所需的所有构建块。PyTorch 中的每个模块都继承自nn.Module。经网络本身就是个模块,它包含其他模块(层)。这种嵌套结构使得构建和管理复杂架构变得容易。

在接下来的部分,我们将构建个经网络来对 FashionMNIST 数据集中的图像进行分类。

获取训练设备#

我们希望能够在加速器(如 CUDA、MPS、MTIA 或 XPU)上训练我们的模型。如果当前加速器可用,我们将使用它。否则,我们将使用 CPU。

定义类#

我们通过继承 nn.Module 来定义经网络,并在 __init__ 中初始化经网络层。每个 nn.Module 子类都在 forward 法中实现对输入数据的操作。

我们创建 NeuralNetwork 的实例,将其移动到 device,并印其结构。

要使用模型,我们会将输入数据传递给它。这会执行模型的 forward 法以及些后台操作。请勿直接调用 model.forward()!

奥力斯    pvc管道管件胶批发    联系人:王经理    手机:15226765735(微信同号)    地址:河北省任丘市北辛庄乡南代河工业区株洲防火门胶

在输入上调用模型会返回个二维张量,其中 dim= 对应于每个类别的 1 个原始预测值的输出,dim=1 对应于每个输出的单个值。通过将输出传递给 nn.Softmax 模块的实例,我们获得预测概率。

模型层#

让我们分解下 FashionMNIST 模型中的层。为了说明这点,我们将取个包含 3 个 28x28 图像的样本小批量,看看它在通过网络时会发生什么。

nn.Flatten#

我们初始化 nn.Flatten 层,将每个 2D 28x28 图像转换为个连续的 784 像素值数组(小批量维度(在 dim= 上)保持不变)。

nn.Linear#

线层 是个模块,它使用其存储的权重和偏差对输入应用线变换。

nn.ReLU#

非线激活会创建模型输入和输出之间的复杂映射。它们在线变换后应用,万能胶厂家以引入*非线*,帮助经网络学习各种现象。

在此模型中,我们在线层之间使用 nn.ReLU,但还有其他激活函数可以在模型中引入非线。

nn.Sequential#

nn.Sequential 是个有序的模块容器。数据按定义的顺序通过所有模块。您可以使用顺序容器快速组网络,例如 seq_modules。

nn.Softmax#

经网络的后个线层返回logits - 原始值在 [-infty, infty] 范围内 - 这些值被传递给nn.Softmax 模块。logits 被缩放到 [, 1] 范围内的值,表示模型对每个类别的预测概率。dim 参数指示了须将值求和为 1 的维度。

模型参数#

经网络中的许多层都是*参数化的*,即它们具有在训练期间化的相关权重和偏差。继承 nn.Module 会自动跟踪模型对象中定义的所有字段,并通过模型的 parameters() 或 named_parameters() 法使所有参数都可访问。

在此示例中,我们迭代每个参数,并印其大小和值的预览。

进步阅读#

torch.nn API

脚本总运行时间: ( 分钟 .56 秒)

相关词条:铝皮保温     隔热条设备     钢绞线厂家玻璃棉    泡沫板橡塑板专用胶

1.本网站以及本平台支持关于《新广告法》实施的“极限词“用语属“违词”的规定,并在网站的各个栏目、产品主图、详情页等描述中规避“违禁词”。
2.本店欢迎所有用户指出有“违禁词”“广告法”出现的地方,并积极配合修改。
3.凡用户访问本网页,均表示默认详情页的描述,不支持任何以极限化“违禁词”“广告法”为借口理由投诉违反《新广告法》,以此来变相勒索商家索要赔偿的违法恶意行为。

Powered by 任丘市奥力斯涂料厂 RSS地图 HTML地图

Copyright Powered by站群系统 © 2025-2054

任丘市奥力斯涂料厂