pytorch学习笔记-神经网络的基本架构（nn.module的使用）

containers

共有以下六个模块

 

import torch
from torch import nn


class kele(nn.Module):
    def __init__(self) -> None:
        super().__init__()
    def forward(self,input):
        output=input+1
        return output
Kele=kele()
x=torch.tensor(1.0)
#将1.0这个数据转化成tensor类型
output=Kele(x)
print(output)
#tensor(2.)