beanmachine.ppl.compiler.bm_graph_builder module

class beanmachine.ppl.compiler.bm_graph_builder.BMGraphBuilder(execution_context: beanmachine.ppl.compiler.execution_context.ExecutionContext = <beanmachine.ppl.compiler.execution_context.ExecutionContext object>)

Bases: object

add_addition(left: beanmachine.ppl.compiler.bmg_nodes.BMGNode, right: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
add_bernoulli(probability: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BernoulliNode
add_bernoulli_logit(probability: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BernoulliLogitNode
add_beta(alpha: beanmachine.ppl.compiler.bmg_nodes.BMGNode, beta: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BetaNode
add_binomial(count: beanmachine.ppl.compiler.bmg_nodes.BMGNode, probability: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BinomialNode
add_binomial_logit(count: beanmachine.ppl.compiler.bmg_nodes.BMGNode, probability: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BinomialLogitNode
add_bitand(left: beanmachine.ppl.compiler.bmg_nodes.BMGNode, right: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
add_bitor(left: beanmachine.ppl.compiler.bmg_nodes.BMGNode, right: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
add_bitxor(left: beanmachine.ppl.compiler.bmg_nodes.BMGNode, right: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
add_boolean(value: bool) beanmachine.ppl.compiler.bmg_nodes.BooleanNode
add_boolean_matrix(value: torch.Tensor) beanmachine.ppl.compiler.bmg_nodes.ConstantBooleanMatrixNode
add_categorical(probability: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.CategoricalNode
add_categorical_logit(probability: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.CategoricalLogitNode
add_chi2(df: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.Chi2Node
add_choice(condition: beanmachine.ppl.compiler.bmg_nodes.BMGNode, *values: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.ChoiceNode
add_cholesky(operand: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
add_column_index(left: beanmachine.ppl.compiler.bmg_nodes.BMGNode, right: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
add_complement(operand: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
add_constant(value: Any) beanmachine.ppl.compiler.bmg_nodes.UntypedConstantNode

This takes any constant value of a supported type, creates a constant graph node for it, and adds it to the builder

add_constant_of_matrix_type(value: Any, node_type: beanmachine.ppl.compiler.bmg_types.BMGMatrixType) beanmachine.ppl.compiler.bmg_nodes.ConstantNode
add_constant_of_type(value: Any, node_type: beanmachine.ppl.compiler.bmg_types.BMGLatticeType) beanmachine.ppl.compiler.bmg_nodes.ConstantNode

This takes any constant value of a supported type and creates a constant graph node of the stated type for it, and adds it to the builder

add_constant_tensor(value: torch.Tensor) beanmachine.ppl.compiler.bmg_nodes.ConstantTensorNode
add_dirichlet(concentration: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.DirichletNode
add_division(left: beanmachine.ppl.compiler.bmg_nodes.BMGNode, right: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
add_equal(left: beanmachine.ppl.compiler.bmg_nodes.BMGNode, right: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
add_exp(operand: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
add_exp2(operand: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
add_exp_product(*inputs: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.ExpProductFactorNode
add_expm1(operand: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
add_flat() beanmachine.ppl.compiler.bmg_nodes.FlatNode
add_floordiv(left: beanmachine.ppl.compiler.bmg_nodes.BMGNode, right: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
add_gamma(concentration: beanmachine.ppl.compiler.bmg_nodes.BMGNode, rate: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.GammaNode
add_greater_than(left: beanmachine.ppl.compiler.bmg_nodes.BMGNode, right: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
add_greater_than_equal(left: beanmachine.ppl.compiler.bmg_nodes.BMGNode, right: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
add_halfcauchy(scale: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.HalfCauchyNode
add_halfnormal(sigma: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.HalfNormalNode
add_if_then_else(condition: beanmachine.ppl.compiler.bmg_nodes.BMGNode, consequence: beanmachine.ppl.compiler.bmg_nodes.BMGNode, alternative: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
add_in(left: beanmachine.ppl.compiler.bmg_nodes.BMGNode, right: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
add_index(left: beanmachine.ppl.compiler.bmg_nodes.BMGNode, right: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
add_invert(operand: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
add_is(left: beanmachine.ppl.compiler.bmg_nodes.BMGNode, right: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
add_is_not(left: beanmachine.ppl.compiler.bmg_nodes.BMGNode, right: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
add_item(input: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
add_less_than(left: beanmachine.ppl.compiler.bmg_nodes.BMGNode, right: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
add_less_than_equal(left: beanmachine.ppl.compiler.bmg_nodes.BMGNode, right: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
add_log(operand: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
add_log10(operand: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
add_log1mexp(operand: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
add_log1p(operand: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
add_log2(operand: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
add_logaddexp(left: beanmachine.ppl.compiler.bmg_nodes.BMGNode, right: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.LogAddExpNode
add_logistic(operand: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
add_logsumexp(*inputs: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.LogSumExpNode
add_logsumexp_torch(input: beanmachine.ppl.compiler.bmg_nodes.BMGNode, dim: beanmachine.ppl.compiler.bmg_nodes.BMGNode, keepdim: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.LogSumExpTorchNode
add_logsumexp_vector(operand: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.LogSumExpVectorNode
add_lshift(left: beanmachine.ppl.compiler.bmg_nodes.BMGNode, right: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
add_matrix_multiplication(left: beanmachine.ppl.compiler.bmg_nodes.BMGNode, right: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
add_matrix_scale(scalar: beanmachine.ppl.compiler.bmg_nodes.BMGNode, matrix: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
add_mod(left: beanmachine.ppl.compiler.bmg_nodes.BMGNode, right: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
add_multi_addition(*inputs: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
add_multi_multiplication(*inputs: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
add_multiplication(left: beanmachine.ppl.compiler.bmg_nodes.BMGNode, right: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
add_natural(value: int) beanmachine.ppl.compiler.bmg_nodes.NaturalNode
add_natural_matrix(value: torch.Tensor) beanmachine.ppl.compiler.bmg_nodes.ConstantNaturalMatrixNode
add_neg_real(value: float) beanmachine.ppl.compiler.bmg_nodes.NegativeRealNode
add_neg_real_matrix(value: torch.Tensor) beanmachine.ppl.compiler.bmg_nodes.ConstantNegativeRealMatrixNode
add_negate(operand: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
add_node(node: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode

This adds a node we’ve recently created to the node set; it maintains the invariant that all the input nodes are also added.

add_normal(mu: beanmachine.ppl.compiler.bmg_nodes.BMGNode, sigma: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.NormalNode
add_not(operand: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
add_not_equal(left: beanmachine.ppl.compiler.bmg_nodes.BMGNode, right: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
add_not_in(left: beanmachine.ppl.compiler.bmg_nodes.BMGNode, right: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
add_observation(observed: beanmachine.ppl.compiler.bmg_nodes.SampleNode, value: Any) beanmachine.ppl.compiler.bmg_nodes.Observation
add_phi(operand: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
add_poisson(rate: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.PoissonNode
add_pos_real(value: float) beanmachine.ppl.compiler.bmg_nodes.PositiveRealNode
add_pos_real_matrix(value: torch.Tensor) beanmachine.ppl.compiler.bmg_nodes.ConstantPositiveRealMatrixNode
add_power(left: beanmachine.ppl.compiler.bmg_nodes.BMGNode, right: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
add_probability(value: float) beanmachine.ppl.compiler.bmg_nodes.ProbabilityNode
add_probability_matrix(value: torch.Tensor) beanmachine.ppl.compiler.bmg_nodes.ConstantProbabilityMatrixNode
add_query(operator: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.Query
add_real(value: float) beanmachine.ppl.compiler.bmg_nodes.RealNode
add_real_matrix(value: torch.Tensor) beanmachine.ppl.compiler.bmg_nodes.ConstantRealMatrixNode
add_rshift(left: beanmachine.ppl.compiler.bmg_nodes.BMGNode, right: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
add_sample(operand: beanmachine.ppl.compiler.bmg_nodes.DistributionNode) beanmachine.ppl.compiler.bmg_nodes.SampleNode
add_simplex(value: torch.Tensor) beanmachine.ppl.compiler.bmg_nodes.ConstantSimplexMatrixNode
add_squareroot(operand: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
add_studentt(df: beanmachine.ppl.compiler.bmg_nodes.BMGNode, loc: beanmachine.ppl.compiler.bmg_nodes.BMGNode, scale: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.StudentTNode
add_subtraction(left: beanmachine.ppl.compiler.bmg_nodes.BMGNode, right: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
add_sum(operand: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
add_switch(*elements: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.SwitchNode
add_tensor(size: torch.Size, *data: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.TensorNode
add_to_int(operand: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
add_to_matrix(rows: beanmachine.ppl.compiler.bmg_nodes.NaturalNode, columns: beanmachine.ppl.compiler.bmg_nodes.NaturalNode, *data: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.ToMatrixNode
add_to_negative_real(operand: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
add_to_positive_real(operand: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
add_to_positive_real_matrix(operand: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
add_to_probability(operand: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
add_to_real(operand: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
add_to_real_matrix(operand: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
add_uniform(low: beanmachine.ppl.compiler.bmg_nodes.BMGNode, high: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.UniformNode
add_vector_index(left: beanmachine.ppl.compiler.bmg_nodes.BMGNode, right: beanmachine.ppl.compiler.bmg_nodes.BMGNode) beanmachine.ppl.compiler.bmg_nodes.BMGNode
all_ancestor_nodes() List[beanmachine.ppl.compiler.bmg_nodes.BMGNode]

Returns a topo-sorted list of nodes that are ancestors to any sample, observation, query or factor.

all_nodes() List[beanmachine.ppl.compiler.bmg_nodes.BMGNode]

Returns a topo-sorted list of all nodes.

all_observations() List[beanmachine.ppl.compiler.bmg_nodes.Observation]
execution_context: beanmachine.ppl.compiler.execution_context.ExecutionContext
remove_leaf(node: beanmachine.ppl.compiler.bmg_nodes.BMGNode) None

This removes a leaf node from the builder, and ensures that the output edges of all its input nodes are removed as well.

remove_node(node: beanmachine.ppl.compiler.bmg_nodes.BMGNode) None

This removes a node from the builder, and ensures that the output edges of all its input nodes are removed as well.

beanmachine.ppl.compiler.bm_graph_builder.phi(x: Any) Any