Parameter作為Module類(lèi)的參數(shù),可以自動(dòng)的添加到Module類(lèi)的參數(shù)列表中,并且可以使用Module.parameters()提供的迭代器獲取到
使用parm.data訪問(wèn)這個(gè)參數(shù)的具體數(shù)據(jù)
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參數(shù),而不是只有nn.Linear