treeflow.model.approximation.iaf module

treeflow.model.approximation.iaf.get_inverse_autoregressive_flow_approximation(model: ~tensorflow_probability.python.distributions.joint_distribution.JointDistribution, hidden_units_per_layer: int, dtype=tf.float64, joint_bijector_func: ~typing.Callable[[~tensorflow_probability.python.distributions.joint_distribution.JointDistribution], ~tensorflow_probability.python.bijectors.composition.Composition] = <function get_default_event_space_bijector>, event_shape_fn: ~typing.Callable[[~tensorflow_probability.python.distributions.joint_distribution.JointDistribution], object] = <function event_shape_fn>, seed=None, n_hidden_layers: int = 2, n_iaf_bijectors: int = 2) → Tuple[Distribution, Dict[str, Variable]]

treeflow.model.approximation.iaf.get_fixed_topology_inverse_autoregressive_flow_approximation(model: JointDistribution, hidden_units_per_layer: int, topology_pins: Dict[str, TensorflowTreeTopology], dtype=tf.float64, n_hidden_layers: int = 2, n_iaf_bijectors: int = 2, init_loc=None) → Tuple[Distribution, Dict[str, Tensor]]