データリスト

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,)