Parameter

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

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

Parameters 是 Tensor 的子類，它們與 Module 一起使用時具有非常特殊的屬性——當它們被賦值為 Module 屬性時，會自動新增到其引數列表中，例如會出現在 parameters() 迭代器中。賦值一個 Tensor 則沒有這種效果。這是因為人們可能想在模型中快取一些臨時狀態，例如 RNN 的最後一個隱藏狀態。如果沒有 Parameter 這樣的類，這些臨時狀態也會被註冊。

引數

• data (Tensor) – 引數 tensor。

• requires_grad (bool, 可選) – 引數是否需要梯度。請注意，torch.no_grad() 上下文不會影響 Parameter 建立的預設行為——即使在 no_grad 模式下，Parameter 仍將具有 requires_grad=True。有關更多詳細資訊，請參閱區域性停用梯度計算。預設值：True