MULTIVAE.inference(x, batch_index, cont_covs, cat_covs, n_samples=1)[source]

Run the inference (recognition) model.

In the case of variational inference, this function will perform steps related to computing variational distribution parameters. In a VAE, this will involve running data through encoder networks.

This function should return a dictionary with str keys and Tensor values.

Return type

{str: Tensor}Dict[str, Tensor]