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