如何使一个batch内加载不同大小的数据

解决:自定义dataloader内的collate_fn函数
做法: batch内大小可变的输入

import torch
from torch.utils.data import DataLoader
from torchvision import transforms
import torchvision.datasets as datasets
import matplotlib.pyplot as plt

# a simple custom collate function, just to show the idea
def