sfaira.models.embedding.ModelVaeIAFVersioned.predict_embedding

ModelVaeIAFVersioned.predict_embedding(x, variational=False, return_z0=False)