torch.nn.utils.parametrize.is_parametrized

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

確定一個模組是否具有引數化。

引數

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

• tensor_name (str, 可選) – 模組中引數的名稱 預設值: None

返回值

如果 module 針對名為 tensor_name 的引數具有引數化，或者當 tensor_name 為 None 時具有任何引數化，則返回 True；否則返回 False

返回型別

bool