pytorch中collate_fn函数

为什么要使用collate_fn

 PyTorch DataLoader工作原理可视化

注意: 在最后一步堆叠的时候可能会出现问题: 如果一条数据中所含有的每个数据元的长度不同, 那么将无法进行堆叠. 如: multi-hot类型的数据, 序列数据.

collate_fn函数最大的一个好处: 可以自定义取出一个batch数据的格式. 该函数的输出就是对dataloader进行遍历, 取出一个batch的数据.

你可能感兴趣的:(pytorch)