Parameter

class torch.nn.parameter.Parameter(data=None, requires_grad=True)

一種被視為模組參數的 Tensor。

Parameters 是 Tensor 的子類，當與 Module 一起使用時，具有非常特殊的屬性 - 當它們被指定為 Module 的屬性時，它們會自動添加到其參數列表中，並且會出現在例如 parameters() 迭代器中。 指定 Tensor 沒有這種效果。 這是因為人們可能想要在模型中緩存一些臨時狀態，例如 RNN 的最後一個隱藏狀態。 如果沒有 Parameter 這樣的類，這些臨時變量也會被註冊。

參數

• data (Tensor) – 參數張量。

• requires_grad (bool, optional) – 如果參數需要梯度。 請注意，torch.no_grad() 上下文不影響 Parameter 創建的預設行為 - Parameter 在 no_grad 模式下仍將具有 requires_grad=True。 有關更多詳細資訊，請參閱局部停用梯度計算。 預設值：True