where
* `pulling` indicates if one of the shape should be pulled upward,
- * `nb_sequences` is the number of sequences to generate
- * `nb_images` is the number of frames per sequence
- * `image_height` and `image_width` is the individual image size
- * `nb_shapes` is the number of rectangular shapes in the simulation
- * `random_shape_size` states if the shapes should be of different sizes
- * `random_colors` states if they should be of different colors
+ * `nb_sequences` is the number of sequences to generate,
+ * `nb_images` is the number of frames per sequence,
+ * `image_height` and `image_width` is the individual image size,
+ * `nb_shapes` is the number of rectangular shapes in the simulation,
+ * `random_shape_size` states if the shapes should be of different sizes,
+ * `random_colors` states if they should be of different colors.
The returned ByteTensor has five dimensions: