神经网络基本框架nn.Module

前言：

在PyTorch中，nn.Module是构建所有神经网络的基础类。几乎所有的神经网络模型都会继承这个类来构建。nn.Module提供了一系列功能和属性，使得模型构建、参数管理、模型保存和加载等操作变得更加简单和直观。

代码实现:

1.导入模块

from torch import nn
import torch

2.编写Module类，继承nn.Module类：

class Module(nn.Module):<br /> def __init__(self):<br /> super().__init__()<br /> def forward(self, input):<br /> output = input + 1<br /> return output

3.实例化Module类对象:

module = Module()
x = torch.tensor(1.0)
output = module(x)
print(output)

完整代码：

# 1.导入模块
from torch import nn

# 2.编写Module类，继承nn.Module类：
class Module(nn.Module):
    def __init__(self):
        super().__init__()
    def forward(self, input):
        output = input + 1
        return output

# 3.实例化Module类对象:
module = Module()
x  = torch.tensor()
output = module(x)
print(output)