buddi.models.components.layers#

Classes

Layer(*args, **kwargs)

This is the class from which all layers inherit.

ReparameterizationLayer(*args, **kwargs)

Custom layer that applies the reparameterization trick.