【小白学习PyTorch教程】三、Pytorch中的NN模块并实现第一个神经网络模型
发布日期:2021-07-01 02:11:23 浏览次数:2 分类:技术文章

本文共 145 字,大约阅读时间需要 1 分钟。

@Author:Runsen

文章目录

在PyTorch实现模型建立,主要是NN模块。

nn.Linear

nn.Linear是创建一个线性层。这里需要将输入和输出维度作为参数传递。

linear = nn.Linear(10, 2)example_input = torch.randn(

转载地址:https://maoli.blog.csdn.net/article/details/117912669 如侵犯您的版权,请留言回复原文章的地址,我们会给您删除此文章,给您带来不便请您谅解!

上一篇:【小白学习PyTorch教程】四、基于nn.Module类实现线性回归模型
下一篇:【小白学习PyTorch教程】二、动态计算图和GPU支持操作

发表评论

最新留言

表示我来过!
[***.240.166.169]2024年05月04日 07时08分02秒