python zip()函數__函數

python zip()函數

    # print(len(data))    batch 個 tuple（torch.float(3,224,224), str）    # print(data[0][0].size())  [3,224,224]    # print(data[0][1].size())  [len]

    images, captions = zip(*data)

    # images = batch個[3,224,224]     # images[0].size())  [3,224,224]

# captions = batch 個str

1.解壓

傳入的data為batch個元祖，即batch個（images，captions）對兒。

 images, captions = zip(*data)

zip(*data) = （batch個images，batch個captions）

2.壓

data = zip(images，captions)