#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,
- int nb_shapes,
+ long nb_shapes,
int random_shape_size, int random_colors) {
long nb_channels = 3;
a = THByteTensor_storage(result)->data + THByteTensor_storageOffset(result) + s * st0;
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++) {