python库中的setter方法和getter方法
setattr方法
setattr
setattr(object, name, value)
- object:要设置属性值的对象。
- name:要设置的属性名。
- value:要设置的属性值。
setattr 函数会将 object 对象中名为 name 的属性的值设置为 value,如果 name 不存在,它会在 object 中创建一个名为 name 的属性,并将其值设置为 value。
举个例子,假设有一个类 Person,它有一个名为 name 的属性,可以通过 setattr 函数来设置 name
class Person:
def __init__(self, name):
self.name = name
person = Person('Tom')
setattr(person, 'name', 'Jerry')
print(person.name) # 输出:'Jerry'
结果
Jerry
上述代码中,首先创建了一个 Person 对象 person,然后使用 setattr 函数将 person 对象的 name 属性的值设置为 'Jerry',最后输出 person 对象的 name 属性的值,结果为 'Jerry'。
另外,也可以使用 setattr
class Person:
pass
person = Person()
setattr(person, 'name', 'Tom')
print(person.name) # 输出:'Tom'
结果
Tom
上述代码中,首先创建了一个空的 Person 对象 person,然后使用 setattr 函数为 person 对象添加一个名为 name 的属性,并将其值设置为 'Tom',最后输出 person 对象的 name 属性的值,结果为 'Tom'。
getter方法
getattr 是 Python 中的一个内置函数,用于获取对象的属性值。它的作用是从一个对象中获取指定名称的属性值,如果该属性不存在,则会抛出一个 AttributeError
getattr(object, name[, default])
- object:要获取属性值的对象。
- name:要获取的属性名。
- default:可选参数,如果指定的属性不存在,则返回 default 指定的默认值,否则抛出 AttributeError 异常。如果不指定 default 参数,则抛出 AttributeError
getattr 函数会从 object 对象中获取名为 name 的属性的值,如果 name 不存在,它会根据 default
举个例子,假设有一个类 Person,它有一个名为 name 的属性,可以通过 getattr 函数来获取 name
class Person:
def __init__(self, name):
self.name = name
person = Person('Tom')
name = getattr(person, 'name')
print(name) # 输出:'Tom'
结果
Tom
上述代码中,首先创建了一个 Person 对象 person,然后使用 getattr 函数获取 person 对象的 name 属性的值,并将其赋值给变量 name,最后输出 name 的值,结果为 'Tom'。
如果尝试获取一个对象中不存在的属性,getattr 函数会抛出一个 AttributeError
class Person:
def __init__(self, name):
self.name = name
person = Person('Tom')
age = getattr(person, 'age')
结果:
Traceback (most recent call last):
File "D:/code/python_code/TSM_pra/main.py", line 6, in <module>
age = getattr(person, 'age')
AttributeError: 'Person' object has no attribute 'age'
上述代码中,尝试获取 person 对象的 age 属性的值,但是 Person 类中并没有定义 age 属性,因此 getattr 函数会抛出一个 AttributeError
pytorch库中的setter方法和getter方法
setter方法
torch.nn.Module 是 PyTorch 框架中一个重要的类,表示神经网络中的模块。在 torch.nn.Module 类中,有一个 setattr 方法,其作用与 Python 内置的 setattr
setattr
- name:属性名。
- value:属性值。
setattr 方法将 name 属性的值设置为 value。需要注意的是,在 torch.nn.Module 中,调用 setattr
举个例子,假设有一个自定义的神经网络模块 MyNet,其中有一个名为 fc 的全连接层。我们可以使用 setattr 方法将 fc 层的权重参数和偏置参数分别设置为 weight 和 bias:
import torch
import torch.nn as nn
class MyNet(nn.Module):
def __init__(self):
super(MyNet, self).__init__()
self.fc = nn.Linear(10, 5)
net = MyNet()
setattr(net.fc, 'weight', torch.randn(5, 10))
setattr(net.fc, 'bias', torch.randn(5))
上述代码中,我们首先定义了一个名为 MyNet 的自定义模块,其中包含一个名为 fc 的全连接层。然后,我们创建了一个 MyNet 对象 net,并使用 setattr 方法设置 fc 层的权重参数和偏置参数,其中 weight 的形状为 (5, 10),bias 的形状为 (5,)。
需要注意的是,在实际使用中,我们通常使用 nn.Module 类的 __setattr__ 方法来设置模块中的属性值,而不是直接调用 setattr 方法。这是因为 __setattr__ 方法会调用 PyTorch 内置的 _parameters 和 _buffers
import torch
import torch.nn as nn
class MyNet(nn.Module):
def __init__(self):
super(MyNet, self).__init__()
self.fc = nn.Linear(10, 5)
def forward(self, x):
x = self.fc(x)
return x
net = MyNet()
net.fc.weight = nn.Parameter(torch.randn(5, 10))
net.fc.bias = nn.Parameter(torch.randn(5))
上述代码中,我们使用 PyTorch 内置的 nn.Parameter 类来将权重参数和偏置参数转换为可训练的参数,然后直接对 fc 层的 weight 和 bias 属性赋值即可。这种方式比直接调用 setattr
getter方法
torch.nn.Module 是 PyTorch 框架中一个重要的类,表示神经网络中的模块。在 torch.nn.Module 类中,有一个 getattr 方法,其作用与 Python 内置的 getattr
getattr
- name:属性名。
getattr 方法返回 name 属性的值。需要注意的是,在 torch.nn.Module 中,调用 getattr
举个例子,假设有一个自定义的神经网络模块 MyNet,其中有一个名为 fc 的全连接层。我们可以使用 getattr 方法获取 fc
import torch
import torch.nn as nn
class MyNet(nn.Module):
def __init__(self):
super(MyNet, self).__init__()
self.fc = nn.Linear(10, 5)
net = MyNet()
weight = getattr(net.fc, 'weight')
bias = getattr(net.fc, 'bias')
上述代码中,我们首先定义了一个名为 MyNet 的自定义模块,其中包含一个名为 fc 的全连接层。然后,我们创建了一个 MyNet 对象 net,并使用 getattr 方法获取 fc 层的权重参数和偏置参数,分别赋值给 weight 和 bias
需要注意的是,如果 getattr 方法获取的属性值是一个子模块,我们可以通过调用子模块的 parameters 方法获取其所有参数,或者通过调用子模块的 children
import torch
import torch.nn as nn
class MyNet(nn.Module):
def __init__(self):
super(MyNet, self).__init__()
self.fc1 = nn.Linear(10, 5)
self.fc2 = nn.Linear(5, 2)
net = MyNet()
fc1_params = list(getattr(net, 'fc1').parameters())
fc2_children = list(getattr(net, 'fc2').children())
上述代码中,我们定义了一个名为 MyNet 的自定义模块,其中包含两个全连接层 fc1 和 fc2。然后,我们创建了一个 MyNet 对象 net,并使用 getattr 方法获取 fc1 层和 fc2 层,分别调用它们的 parameters 和 children