pytorch dataloader collate_fn

pytorch中的dataloader collate_fn是用来将多个样本合并为一个batch的函数。通常情况下，它会将多个样本直接堆叠在一起，但是如果每个样本的形状不一致，则需要定义一个collate_fn函数来处理。这个函数会根据最大的样本形状来加工数据，使得每个batch中的数据形状一致。