X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=flatland.git;a=blobdiff_plain;f=flatland.h;h=cdde100742f334fa1a4554c101b97d5124d3a4c0;hp=a0a255f89cd6ab542bd3c63f6cb91a8b7ba0f9be;hb=4a824920ad843ba0d1bd3d9b10a8eb0188a2bc63;hpb=5d4e9eaeec9263692d39ca840e498a5f1d818eaa diff --git a/flatland.h b/flatland.h index a0a255f..cdde100 100644 --- a/flatland.h +++ b/flatland.h @@ -1,2 +1,7 @@ -int generate_sequence(long nb_sequences, THByteTensor *output); +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);