Update.

[flatland.git] / flatland.h

diff --git a/flatland.h b/flatland.h
index 9a9247a..cdde100 100644 (file)
--- a/flatland.h
+++ b/flatland.h
@@ -1,14 +1,7 @@
 
-#ifdef __cplusplus
-extern "C" {
-#endif
-
-void fl_generate_sequences(int nb_sequences,
-                           int nb_images_per_sequence,
-                           int width, int height,
-                           unsigned char *output);
-
-#ifdef __cplusplus
-}
-#endif
-
+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);