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