torch.nn.utils.parametrize.is_parametrized

torch.nn.utils.parametrize.is_parametrized(module, tensor_name=None)

判斷模組是否具有參數化 (parametrization)。

參數

• module (nn.Module) – 要查詢的模組

• tensor_name (str, optional) – 模組中參數的名稱。預設值： None

回傳

如果 module 具有名為 tensor_name 的參數的參數化，或者當 tensor_name 為 None 時，它具有任何參數化，則回傳 True；否則回傳 False

回傳類型

bool