ComfyUI 데이터 목록과 목록 처리

class ImageRebatch:
    @classmethod
    def INPUT_TYPES(s):
        return {"required": { "images": ("IMAGE",),
                              "batch_size": ("INT", {"default": 1, "min": 1, "max": 4096}) }}
    RETURN_TYPES = ("IMAGE",)
    INPUT_IS_LIST = True
    OUTPUT_IS_LIST = (True, )
    FUNCTION = "rebatch"
    CATEGORY = "image/batch"

    def rebatch(self, images, batch_size):
        batch_size = batch_size[0]    # 모든 입력이 리스트로 오기 때문에 batch_size는 list[int]

        output_list = []
        all_images = []
        for img in images:                    # 각 img는 이미지 배치입니다
            for i in range(img.shape[0]):     # 각 i는 단일 이미지입니다
                all_images.append(img[i:i+1])

        for i in range(0, len(all_images), batch_size): # batch_size만큼의 조각을 떼어 각각 새로운 배치로 만듭니다
            output_list.append(torch.cat(all_images[i:i+batch_size], dim=0))  # 이미지 배치의 너비나 높이가 다르면 심각한 오류 발생!

        return (output_list,)