Module flowcon.CNF.neural_odes.diffeq_layers.wrappers

Functions

def diffeq_wrapper(layer)
def reshape_wrapper(input_shape, layer)