def arrange_array_size(batch_size: int, array_size: tuple):
    """
    Decide array size given batch size and array size [batch_size, height, width, channel].
    :param batch_size: batch size
    :param array_size: array (img) size
    :return: final array size
    """
    output = list(array_size)
    output.insert(0, batch_size)
    return output