小土堆-pytorch-神经网络的基本骨架-nn.Module的使用03_笔记

Moudule基本模块

 向前传播函数

 

实战-书写自己的模块  

# 定义自己的模块
import torch
from torch import nn


class Tudui(nn.Module):
    # 重写初始化模块 左上角->Code->Generate->Ovveride methods
    def __init__(self):
        super().__init__()
    # 重写前向传播函数
    def forward(self,input):
        output=input+1
        return output


# 创建神经网络
tudui=Tudui()
x=torch.tensor(1.0)
output=tudui(x)
print(output)

运行结果:

关于pycharm的调式技巧:

第一步:设置断点

 

 第二步:点击虫子

 第三步:一步一步调试