--batch=shapes_1d
--batch=shapes_2d_small
--batch=shapes_3d_small

