Networks

networks.ConditionalVelocityField(...[, ...])	Parameterized neural vector field with conditions.
networks.GENOTConditionalVelocityField(...)	Parameterized neural vector field with conditions for GENOT.
networks.ConditionEncoder(output_dim[, ...])	Encoder for conditions represented as sets of perturbations.
networks.SelfAttention([num_heads, qkv_dim, ...])	Self-attention layer
networks.SeedAttentionPooling([num_heads, ...])	Pooling by multi-head attention with a trainable seed.
networks.TokenAttentionPooling([num_heads, ...])	Multi-head attention which aggregates sets by learning a token.
networks.MLPBlock([dims, dropout_rate, ...])	MLP block.
networks.SelfAttentionBlock(num_heads, qkv_dim)	Several self-attention (+ optional FC layer) layers stacked together.
networks.BaseModule([parent, name])	Base module for condition encoder and its components.
networks.FilmBlock(input_dim, cond_dim[, ...])	Feature-wise Linear Modulation (FiLM) layer.
networks.ResNetBlock(input_dim[, ...])	Residual conditioning block.
networks.SelfAttentionBlock(num_heads, qkv_dim)	Several self-attention (+ optional FC layer) layers stacked together.