Python CoerceDistributionToFunsor示例

示例#1

文件： handlers.py 项目： xidulu/numpyro

 def __init__(self, *args, **kwargs):
     if collapse._coerce is None:
         import funsor
         from funsor.distribution import CoerceDistributionToFunsor
         funsor.set_backend("jax")
         collapse._coerce = CoerceDistributionToFunsor("jax")
     super().__init__(*args, **kwargs)

示例#2

 def __init__(self, *args, **kwargs):
     if CollapseMessenger._coerce is None:
         import funsor
         from funsor.distribution import CoerceDistributionToFunsor
         funsor.set_backend("torch")
         CollapseMessenger._coerce = CoerceDistributionToFunsor("torch")
     self._block = False
     super().__init__(*args, **kwargs)