Embedding nets

Embedding nets#

sbi.neural_nets.embedding_nets.CausalCNNEmbedding

Embedding network that uses 1D causal convolutions.

sbi.neural_nets.embedding_nets.CNNEmbedding

Convolutional embedding network (1D or 2D convolutions).

sbi.neural_nets.embedding_nets.FCEmbedding

Fully-connected multi-layer neural network to be used as embedding network.

sbi.neural_nets.embedding_nets.LRUEmbedding

Embedding network backed by a stack of Linear Recurrent Unit (LRU) blocks.

sbi.neural_nets.embedding_nets.PermutationInvariantEmbedding

Permutation invariant embedding network.

sbi.neural_nets.embedding_nets.ResNetEmbedding1D

Residual neural network mapping vector-like data to a fixed-size vector.

sbi.neural_nets.embedding_nets.ResNetEmbedding2D

Residual neural network mapping image-like data to a fixed sized vector.

sbi.neural_nets.embedding_nets.SpectralConvEmbedding

sbi.neural_nets.embedding_nets.TransformerEmbedding

Transformer-based embedding network for time series and image data.