【Pytorch】简析DataLoader中的collate_fn参数

如在博文数据批量化处理类Dataset和DataLoader中所介绍的一样,DataLoader可通过collate_fn参数,对Dataset生成的mini-batch的可迭代数据进行进一步处理,而本文就简要介绍下该参数,并给出一个简单的例子。 1. collate_fn的设置、输入和输出 collate_fn应当是一个可调用对象,常见的可以是外部定义的函数或者lambda函数。其接受DataLoader不设置collate_fn参数时每个mini-batch输出的样本序列,所以可以理解为一个后处理的函数。 在不设置collate_fn参数时,DataLoader的mini-batch样本序