Funsor 是一个用于函数和分布的类张量库¶
Funsor 核心
- 操作
- 域
- 解释器
- Funsors
- 优化器
- 伴随算法
- 求和-乘积算法
partial_unroll()partial_sum_product()dynamic_partial_sum_product()modified_partial_sum_product()sum_product()naive_sequential_sum_product()sequential_sum_product()mixed_sequential_sum_product()naive_sarkka_bilmes_product()sarkka_bilmes_product()MarkovProductMetaMarkovProducteager_markov_product()
- 仿射模式匹配
- Funsor 工厂
- 测试工具
xfail_if_not_implemented()xfail_if_not_found()requires_backend()excludes_backend()ActualExpectedid_from_inputs()is_array()assert_close()check_funsor()xfail_param()make_einsum_example()assert_equiv()rand()randint()randn()random_scale_tril()zeros()ones()empty()random_tensor()random_gaussian()random_mvn()make_plated_hmm_einsum()make_chain_einsum()make_hmm_einsum()iter_subsets()DesugarGetitem
- 类型提示工具
接口
- 使用 Funsor 的食谱
forward_filter_backward_rsample()forward_filter_backward_precondition()- Pyro 兼容分布
- 分布 Funsors
DistributionBetaCauchyChi2BernoulliProbsBernoulliLogitsBinomialCategoricalCategoricalLogitsDeltaDirichletDirichletMultinomialExponentialGammaGammaPoissonGeometricGumbelHalfCauchyHalfNormalLaplaceLogisticLowRankMultivariateNormalMultinomialMultivariateNormalNonreparameterizedBetaNonreparameterizedDirichletNonreparameterizedGammaNonreparameterizedNormalNormalParetoPoissonStudentTUniformVonMises
- Mini-Pyro 接口
- Mini Pyro
Distributionget_param_store()MessengertracereplayblockseedCondIndepStackFramePlateMessengertensor_to_funsor()log_jointapply_stack()sample()param()plate()PyroOptimAdamClippedAdamSVIExpectation()elbo()ELBOTrace_ELBOTraceMeanField_ELBOTraceEnum_ELBOJitJit_ELBOJitTrace_ELBO()JitTraceMeanField_ELBO()JitTraceEnum_ELBO()
- Einsum 接口
- 编译器和跟踪器