Python PyTorch ConstraintRegistry.register用法及代碼示例

本文簡要介紹python語言中 torch.distributions.constraint_registry.ConstraintRegistry.register 的用法。

用法:

register(constraint, factory=None)

參數：

• constraint(子類Constraint) -Constraint 的子類，或所需類的單例對象。

• factory(可調用的) -輸入約束對象並返回 Transform 對象的可調用對象。

在此注冊表中注冊Constraint 子類。用法：

@my_registry.register(MyConstraintClass)
def construct_transform(constraint):
    assert isinstance(constraint, MyConstraint)
    return MyTransform(constraint.arg_constraints)