from modules.layers.SamplingLayer import SamplingLayer from modules.layers.VariationalLossLayer import VariationalLossLayer