default_collate_fn

paddle.io. default_collate_fn ( )

paddle.io.DataLoader 中默认组批次函数,输入 batch 为样本列表,格式如下:

[[filed1,filed2,...],[filed1,filed2,...],...]

此函数将其各按字段组装整合为如下批次数据:

[batch_filed1,batch_filed2,...]

参数:
  • batch (list of numpy array|paddle.Tensor) - 批次数据,格式为样本列表,每个样本包含多个字段,字段为numpy array或Tensor

返回:转换为Tensor的批次数据

返回类型: list of paddle.Tensor

用法见 DataLoader