pytorch中的參數類——torch.nn.parameter

1.torch.nn.parameter概要

pytorch官網對torch.nn.parameter的描述如下。

torch.nn.parameter是一個被用作神經網絡模塊參數的tensor。這是一種tensor的子類。
parameters是張量的子類，當與模塊s一起使用時，它們有一個非常特殊的屬性——當它們被重新分配爲模塊屬性時，它們會自動地添加到它的參數列表中，並且會出現在Parameters()迭代器中。分配一個張量沒有這樣的效果。這是因爲可能希望在模型中緩存一些臨時狀態，比如RNN的最後一個隱藏狀態。如果沒有Parameter這樣的類，這些臨時人員也會被註冊。