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 建立的預設行為——在 no_grad 模式下，Parameter 仍將具有 requires_grad=True。有關更多詳細資訊，請參閱 區域性停用梯度計算。預設為 True