THByteTensor *generate_sequence(long nb_sequences,
long nb_images,
long image_height, long image_width,
+ int 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,
tmp_buffer);
unsigned char *r = tmp_buffer;