aeppl.mixture#

Functions

expand_indices(indices, shape)

Convert basic and/or advanced indices into a single, broadcasted advanced indexing operation.

get_stack_mixture_vars(node)

Extract the mixture terms from a *Subtensor* applied to stacked MeasurableVariables.

is_newaxis(x)

logprob_MixtureRV(op, values, *inputs[, name])

param inputs

rv_pull_down(x[, dont_touch_vars])

Pull a RandomVariable Op down through a graph, when possible.

Classes

MixtureRV(indices_end_idx, out_dtype, ...)

A placeholder used to specify a log-likelihood for a mixture sub-graph.