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 (sequence) – 表示引數化的模組序列

• original (Parameter or Tensor) – 被引數化的引數或緩衝區

• unsafe (bool) – 一個布林標誌，表示引數化是否可能改變張量的資料型別 (dtype) 和形狀 (shape)。預設值: False 警告: 註冊時不會檢查引數化的一致性。啟用此標誌風險自擔。

right_inverse(value)

按相反的註冊順序呼叫引數化的 right_inverse 方法。

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

引數

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