zea.models.layers

Layers used in zea.models

Functions

DownBlock(width, block_depth)	Downsampling block with residual connections.
ResidualBlock(width)	Residual block with swish activation.
UpBlock(width, block_depth)	Upsampling block with residual connections.
sinusoidal_embedding(x, ...)	Sinusoidal embedding layer.

zea.models.layers.DownBlock(width, block_depth)

Downsampling block with residual connections.

zea.models.layers.ResidualBlock(width)

Residual block with swish activation.

zea.models.layers.UpBlock(width, block_depth)

Upsampling block with residual connections.

zea.models.layers.sinusoidal_embedding(x, embedding_min_frequency, embedding_max_frequency, embedding_dims)

Sinusoidal embedding layer.