当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


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)

相关用法


注:本文由纯净天空筛选整理自pytorch.org大神的英文原创作品 torch.distributions.constraint_registry.ConstraintRegistry.register。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。