Mini-Pyro 接口

此接口为 Pyro 概率编程语言提供后端。该接口旨在通过编写标准 Pyro 代码并设置 pyro_backend("funsor") 来间接使用。有关示例用法，请参阅 examples/minipyro.py。

Mini Pyro

此文件包含 Pyro 概率编程语言的最小实现。其 API（方法签名等）尽可能与完整实现相匹配。除 pyro.distributions 模块外，此文件独立于 Pyro 的其余部分。

一个使用此实现的配套示例可在 examples/minipyro.py 中找到。

class Distribution(funsor_dist, sample_inputs=None)

基类： object

log_prob(value)

expand_inputs(name, size)

get_param_store()

class Messenger(fn=None)

基类： object

process_message(msg)

postprocess_message(msg)

class trace(fn=None)

基类： Messenger

postprocess_message(msg)

get_trace(*args, **kwargs)

class replay(fn, guide_trace)

基类： Messenger

process_message(msg)

class block(fn=None, hide_fn=<function block.<lambda>>)

基类： Messenger

process_message(msg)

class seed(fn=None, rng_seed=None)

基类： Messenger

class CondIndepStackFrame(name, size, dim)

基类： tuple

property dim

字段编号 2 的别名

property name

字段编号 0 的别名

property size

字段编号 1 的别名

class PlateMessenger(fn, name, size, dim)

基类： Messenger

process_message(msg)

tensor_to_funsor(value, cond_indep_stack, output)

class log_joint(fn=None)

基类： Messenger

process_message(msg)

postprocess_message(msg)

apply_stack(msg)

sample(name, fn, obs=None, infer=None)

param(name, init_value=None, constraint=Real(), event_dim=None)

plate(name, size, dim)

class PyroOptim(optim_args)

基类： object

class Adam(optim_args)

基类： PyroOptim

TorchOptimizer

的别名 Adam

class ClippedAdam(optim_args)

基类： PyroOptim

TorchOptimizer

的别名 ClippedAdam

class SVI(model, guide, optim, loss)

基类： object

step(*args, **kwargs)

Expectation(log_probs, costs, sum_vars, prod_vars)

elbo(model, guide, *args, **kwargs)

class ELBO(**kwargs)

基类： object

class Trace_ELBO(**kwargs)

基类： ELBO

class TraceMeanField_ELBO(**kwargs)

基类： ELBO

class TraceEnum_ELBO(**kwargs)

基类： ELBO

class Jit(fn, **kwargs)

基类： object

class Jit_ELBO(elbo, **kwargs)

基类： ELBO

JitTrace_ELBO(**kwargs)

JitTraceMeanField_ELBO(**kwargs)

JitTraceEnum_ELBO(**kwargs)