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.)