神经网络-nn.Module

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模块详解
标记下,以后忘了自己好看

上一篇:jsp:forward相当于requestDispatcher还是sendRedirect呢?


下一篇:python turtle 各种示例