from torch import nn
import torch
class Jiaxin(nn.Module):
def __init__(self):#可以自己写也可以在Code-- generate--重写中选择
super(Jiaxin,self).__init__()#调用父类初始化函数
def forward(self,input):
output=input+1#自定义
return output
jiaxin=Jiaxin()#实例化
x=torch.tensor(1.0)
output=jiaxin(x)#forward是用__call__的方式实现
print(output)
output
#tensor(2.)
【无标题】
最新推荐文章于 2024-01-18 23:25:36 发布