天天看點

nn.parameter

class

torch.nn.parameter.

Parameter

[source]

作為一種模型參數的張量。參數是張量的子類,在與子產品s一起使用時具有一個非常特殊的特性——當它們被指定為子產品屬性時,它們會自動添加到參數清單中,并且會出現,例如在

parameters()疊代器中。

配置設定一個張量就沒有這樣的效果。這是因為人們可能想要在模型中緩存一些臨時狀态,比如RNN的最後一個隐藏狀态。如果沒有像Parameter的類,這些臨時變量也要注冊。

參數: