@@ -12,24 +12,24 @@ class Sequential(keras.Layer):
1212
1313 This class extends `keras.Layer` and provides functionality for building,
1414 calling, and serializing a sequence of layers. Unlike `keras.Sequential`,
15- this implementation allows for more flexibility in handling layer arguments
16- and supports custom serialization through the `@serializable` decorator .
15+ this implementation does not eagerly check input shapes, meaning it is
16+ compatible with both single inputs and sets .
1717
1818 Parameters
1919 ----------
20- layers : keras.Layer or Sequence[keras.Layer]
21- A single Keras layer or a sequence of Keras layers to be managed by this model.
22- **kwargs : dict
20+ layers : keras.layer | Sequence[keras.layer]
21+ A sequence of Keras layers to be managed by this model.
22+ Can be passed by unpacking or as a single sequence.
23+ **kwargs :
2324 Additional keyword arguments passed to the base `keras.Layer` class.
2425
2526 Notes
2627 -----
27- - This class differs from `keras.Sequential` in that it does not assume a strict
28- linear stack of layers and provides custom methods for serialization and
29- configuration.
30- - It is designed to integrate with the BayesFlow framework and supports
31- additional utilities like `layer_kwargs`.
28+ - This class differs from `keras.Sequential` in that it does not eagerly check
29+ input shapes. This means that it is compatible with both single inputs
30+ and sets.
3231 """
32+
3333 def __init__ (self , * layers : keras .Layer | Sequence [keras .Layer ], ** kwargs ):
3434 super ().__init__ (** layer_kwargs (kwargs ))
3535 if len (layers ) == 1 and isinstance (layers [0 ], Sequence ):
0 commit comments