神经网路骨架的搭建nn.Module
torch.NN: neural network
class torch.nn.Module
#官方文档
class Model(nn.Module):
def __init__(self):
super(Model,self).__init__()
self.conv1 = nn.Con2d(1,20,5)
self.conv2 = nn.Con2d(20,20,5)
def forward(self, x):
x = F.relu(self.conv1(x))
return F.relu(self.conv2(x))
#实例:
from torch import nn
class Model_test(nn.Module):
def __init__(self):
super(Model_test, self).__init__()
def forward(self, input)
output = input +1
return output
model_test = Model_test()
x = torch.tensor(1.0)
output = model_test (x)
print(output)
#
class grandfather(): #超类
def __init__(self,name,age):
self.name = name #爷爷的属性
self.age = age
self.height = 180 #爷爷特有的属性
def plan(self):
print("爷爷的计划是环游世界")
def hobby(self):
print("爷爷最喜欢浪遏飞舟")
class father(grandfather): #父类
def __init__(self,name,age):
super().__init__(name,age) #super主要用于集成父类的方法,就是他的上一辈
self.name = name
self.age = age
def diet(self):
print("父亲最喜欢小鸡炖蘑菇")
def hobby(self):
print("父亲最喜欢大美女")
def amazing(self):
super().plan() #super其实就相当于一个类,代表的父辈,可以直接用它调用父辈的函数
class son(father): #子类
def __init__(self,name,age):
super().__init__(name,age) #super方法一般写在类的初始化__init__()的下面,当你实例化son时,他的父类的方法自然而然的继承了进来
self.name = name
self.age = age
def game(self):
print("儿子最喜欢和平精英")
zhen1 = son("xiao_min",20)
print(zhen1.name) #xiao_min //测试儿子的属性
zhen1.game() #儿子最喜欢和平精英 // 测试儿子特有的方法
zhen1.hobby() #父亲最喜欢大美女 //测试儿子继承他父类的方法
zhen1.diet() #父亲最喜欢小鸡炖蘑菇 //测试儿子继承他父类的方法
zhen1.amazing() #爷爷的计划是环游世界 //儿子集成他父亲的amazing()方法,父亲又继承爷爷的plan()方法
print(zhen1.height) #180 //测试儿子类是否可以继承爷爷类的特有属性:可以
zhen1.plan() #爷爷的计划是环游世界 //测试儿子类是否可以直接继承爷爷类的方法:可以
zhen2 = father("da_ming",40)
zhen2.hobby() #父亲最喜欢大美女 //测试父亲与爷爷有相同爱好时怎么办?-------因为zhen2是父亲类,所以父亲类优先
zhen2.plan() #爷爷的计划是环游世界 // 测试当父亲类没有plan()方法时怎们办:程序会一直往年龄大的方向即爷爷类上找
print(zhen2.height) # 180 //测试父亲类是否可以继承爷爷类的独有属性:身高------当然可以
zhen3 = grandfather("lao_ming",90)
#总之一句话,子类优先,子类没有是一直往年龄大的方向找,年龄大的也没就芭比Q了,不仅可以继承类方法,类属性同样可以继承
这篇博客介绍了如何使用PyTorch构建神经网络模型,展示了nn.Module的基本用法,并通过一个简单的例子说明了类继承的概念。在PyTorch中,定义了一个包含两个卷积层的Model类,并实现了前向传播函数。此外,还通过一个类继承的例子,展示了如何创建一个类层次结构,从grandfather到father再到son,每个类都继承并扩展了上一代的属性和方法。
2024

被折叠的 条评论
为什么被折叠?



