#include "sequence_generator.h"
-THByteTensor *generate_sequence(long nb_sequences,
+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) {
long nb_channels = 3;
for(s = 0; s < nb_sequences; s++) {
a = THByteTensor_storage(result)->data + THByteTensor_storageOffset(result) + s * st0;
- fl_generate_sequence(nb_images, image_width, image_height,
+ fl_generate_sequence(nb_images, image_width, image_height, nb_shapes,
random_shape_size, random_colors,
+ pulling,
tmp_buffer);
unsigned char *r = tmp_buffer;
for(k = 0; k < nb_images; k++) {