Pytorch.utils.data.DataLoader中collate_fn参数的使用

来自官网的DataLoader说明： 翻译：合并样本列表中的样本以形成一个最小batch的张量，从映射函数样式数据集中使用批量加载时使用。 个人理解呢，就是从样本列表中过来了一个batch的数据，经过映射函数，形成一个tensor。 collate_fn这玩应就是一个映射函数，输入是一个batch的样本，输出是一个tensor。有系统默认的，当然这个位置作为一个参数被暴露出来的意思就是你可以自己定

>>阅读原文<<