cellflow.networks.GENOTConditionalVelocityField.get_condition_embedding

GENOTConditionalVelocityField.get_condition_embedding(condition)

Get the embedding of the condition.

Parameters:

condition (dict[str, Array]) – Conditioning vector of shape [batch, ...].

Return type:

tuple[Array, Array]

Returns:

: Learnt mean and log-variance of the condition embedding. If cellflow.model.CellFlow.condition_mode is 'deterministic', the log-variance is set to zero.