Minipyro

Mini Pyro

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

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

class Adam(optim_args)

基类: object

__call__(params)

class JitTrace_ELBO(**kwargs)

基类: object

__call__(model, guide, *args)

class Messenger(fn=None)

基类: object

__call__(*args, **kwargs)

postprocess_message(msg)

process_message(msg)

class PlateMessenger(fn, size, dim)

基类: pyro.contrib.minipyro.Messenger

process_message(msg)

class SVI(model, guide, optim, loss)

基类: object

step(*args, **kwargs)

Trace_ELBO(**kwargs)

apply_stack(msg)

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

基类: pyro.contrib.minipyro.Messenger

process_message(msg)

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

get_param_store()

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

plate(name, size, dim=None)

class replay(fn, guide_trace)

基类: pyro.contrib.minipyro.Messenger

process_message(msg)

sample(name, fn, *args, **kwargs)

class seed(fn=None, rng_seed=None)

基类: pyro.contrib.minipyro.Messenger

class trace(fn=None)

基类: pyro.contrib.minipyro.Messenger

get_trace(*args, **kwargs)

postprocess_message(msg)