class CustomDataset(Dataset):
    def __init__(self):
        self.x_data = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
        self.y_data = [[12], [18], [11]]
        def __len__(self):
            return len(self.x_data)
        def __getitem__(self, idx):
            x = torch.FloatTensor(self.x_data[idx])
            y = torch.FloatTensor(self.y_data[idx])
            return x, y
dataset = CustomDataset()
dataloader = DataLoader(
    dataset, ------ 데이터셋
    batch_size=2, ------ 미니 배치 크기로 2의 제곱수를 사용하겠다는 의미입니다.
    shuffle=True, ------ 데이터를 불러올 때마다 랜덤으로 섞어서 가져옵니다.
)