scvi.external.gimvi.JVAE.inference

JVAE.inference(x, mode=None)[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

dictdict