Merge pull request #345 from turnmanh/patch-1

Small doc string improvement.
april-tools · Feb 5, 2025 · 0cd00d8 · 0cd00d8
2 parents 134cd5d + b200550
commit 0cd00d8
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/cirkit/templates/data_modalities.py b/cirkit/templates/data_modalities.py
@@ -40,7 +40,8 @@ def image_data(
         input_layer: The name of the input layer. It can be one of the following:
             'categorical' (encoding a Categorical distribution over pixel channel values),
             'binomial' (encoding a Binomial distribution over pixel channel values),
-            'embedding' (encoding an Embedding vector over pixel channel values).
+            'embedding' (encoding an Embedding vector over pixel channel values),
+            'gaussian' (encoding a Gaussian distribution over pixel channel values).
         num_input_units: The number of input units per input layer.
         sum_product_layer: The name of the sum-product inner layer. It can be one of the following:
             'cp' (the canonical decomposition layer, consisting of dense layers followed by a