ParametrizationList

class torch.nn.utils.parametrize.ParametrizationList(modules, original, unsafe=False)

一個序列容器，用於保存和管理參數化 torch.nn.Module 的原始參數或緩衝區。

當 module[tensor_name] 已使用 register_parametrization() 進行參數化時，它是 module.parametrizations[tensor_name] 的類型。

如果第一個註冊的參數化具有返回一個張量的 right_inverse，或者沒有 right_inverse（在這種情況下，我們假設 right_inverse 是恆等式），它將以名稱 original 保存張量。 如果它有一個返回多個張量的 right_inverse，這些張量將被註冊為 original0、original1、...

警告

此類別由 register_parametrization() 在內部使用。 這裡的文檔僅為完整起見。 不應由使用者實例化。

參數

• modules (序列) – 表示參數化的模組序列

• original (Parameter 或 Tensor) – 參數化後的參數或緩衝區

• unsafe (bool) – 一個布林值標誌，表示參數化是否可能更改張量的 dtype 和形狀。 預設值：False 警告：在註冊時不會檢查參數化的一致性。 啟用此標誌的風險由您自行承擔。

right_inverse(value)

以相反的註冊順序呼叫參數化的 right_inverse 方法。

然後，如果 right_inverse 輸出一個張量，則將結果儲存在 self.original 中；如果輸出多個張量，則儲存在 self.original0、self.original1、... 中。

參數

value (Tensor) – 用於初始化模組的值