Parameter作为Module类的参数,可以自动的添加到Module类的参数列表中,并且可以使用Module.parameters()提供的迭代器获取到
使用parm.data访问这个参数的具体数据
class Net(nn.Module):
def __init__(self):
super().__init__()
self.layer = nn.Linear(1, 1)
self.fun_param = nn.Parameter(torch.FloatTensor([1, ]))
net = Net()
for param in net.named_parameters():
print(param)
print("*"*50)
其中的pararm中就可以含有自定义的fun_param参数,而不是只有nn.Linear