神经网络-nn.Module

2022-01-30 18:59:16 阅读：161 来源： 互联网

torcn.nn是专门为神经网络设计的模块化接口. nn构建于autograd之上，可以用来定义和运行神经网络。
nn.Module是所有网络层基类，用来管理网络属性。所有的网络模型比如说LeNet模型，包括模型的所有子模块，卷积层、池化层也是一个nn.Module类

input—>神经网络（forward）–>output
下面写个示例来复习下forward()

import torch
from torch import nn

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

lwx=Lwx()
x=torch.tensor(1.0)
y=lwx(x)
# tensor(2.)
print(y)

在上组示例中，当调用lwx(x)时，系统会自动调用forward()函数并返回
有个关于nn.module模块写的不错
Pytorch nn.Module模块详解
标记下，以后忘了自己好看

标签：__,Lwx,nn,Module,神经网络,forward
来源： https://blog.csdn.net/qq_52237775/article/details/122754817

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

神经网络-nn.Module