X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;ds=sidebyside;f=flatland.h;h=ba91f6790df022d9d69ddd78495e138f911c81b3;hb=7e2780052c458a329fc36253b83eb5194ffb7fed;hp=a0a255f89cd6ab542bd3c63f6cb91a8b7ba0f9be;hpb=5d4e9eaeec9263692d39ca840e498a5f1d818eaa;p=flatland.git

diff --git a/flatland.h b/flatland.h
index a0a255f..ba91f67 100644
--- a/flatland.h
+++ b/flatland.h
@@ -1,2 +1,6 @@
 
-int generate_sequence(long nb_sequences, THByteTensor *output);
+THByteTensor *generate_sequence(long nb_sequences,
+                                long nb_images,
+                                long image_height, long image_width,
+                                int nb_shapes,
+                                int random_shape_size, int random_colors);