Merge pull request #101 from stefanradev93/Development

Add summary_dim attribute to transformers

Author: stefanradev93

bayesflow/summary_networks.py

@@ -148,6 +148,7 @@ def __init__(
 
         # Final output reduces representation into a vector of length summary_dim
         self.output_layer = Dense(summary_dim)
+        self.summary_dim = summary_dim
 
     def call(self, x, **kwargs):
         """Performs the forward pass through the transformer.
@@ -269,6 +270,8 @@ def __init__(
             summary_dim, attention_settings, num_dense_fc, dense_settings, use_layer_norm, num_seeds
         )
 
+        self.summary_dim = summary_dim
+
     def call(self, x, **kwargs):
         """Performs the forward pass through the set-transformer.