nb_squares = torch.randint(max_nb_squares, (1,)) + 1
square_position = torch.randperm(height * width)[:nb_squares]
nb_squares = torch.randint(max_nb_squares, (1,)) + 1
square_position = torch.randperm(height * width)[:nb_squares]
square_c = torch.randperm(nb_colors)[:nb_squares] + 1
square_i = square_position.div(width, rounding_mode = 'floor')
square_j = square_position % width
square_c = torch.randperm(nb_colors)[:nb_squares] + 1
square_i = square_position.div(width, rounding_mode = 'floor')
square_j = square_position % width