X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=flatland.git;a=blobdiff_plain;f=flatland.h;h=cdde100742f334fa1a4554c101b97d5124d3a4c0;hp=734add49c5de52618fc192d2b54e2adad3806f5f;hb=90d1c5704c30e7f1d041e32eacbc2893741110e1;hpb=4aed0ce274b7c0e379651c28e439375c821c047a

diff --git a/flatland.h b/flatland.h
index 734add4..cdde100 100644
--- a/flatland.h
+++ b/flatland.h
@@ -1,4 +1,7 @@
 
-THByteTensor *generate_sequence(long nb_sequences,
-                                long nb_images_per_sequence,
-                                long image_width, long image_height);
+THByteTensor *generate_sequence(int pulling,
+                                long nb_sequences,
+                                long nb_images,
+                                long image_height, long image_width,
+                                long nb_shapes,
+                                int random_shape_size, int random_colors);