SqueezeMixin#

class sinabs.layers.SqueezeMixin[source]#

Utility mixin class that will wrap the __init__ and forward call to flatten the input to and the output from a child class.

The wrapped __init__ will provide two additional parameters batch_size and num_timesteps and the wrapped forward will unpack and repack the first dimension into batch and time.