Update.

[flatland.git] / flatland.h

diff --git a/flatland.h b/flatland.h
index a0a255f..734add4 100644 (file)
--- a/flatland.h
+++ b/flatland.h
@@ -1,2 +1,4 @@
 
-int generate_sequence(long nb_sequences, THByteTensor *output);
+THByteTensor *generate_sequence(long nb_sequences,
+                                long nb_images_per_sequence,
+                                long image_width, long image_height);