pytorch.utils.data.DataLoader:将自定义的Dataset根据batch size大小、是否shuffle等封装成一个Batch Size大小的Tensor,用于后面的训练,dataloader本质是一个可迭代对象,使用iter()访问,不能使用next()访问。
PyTorch数据处理工具箱utils
最新推荐文章于 2023-08-20 09:34:54 发布
pytorch.utils.data.DataLoader:将自定义的Dataset根据batch size大小、是否shuffle等封装成一个Batch Size大小的Tensor,用于后面的训练,dataloader本质是一个可迭代对象,使用iter()访问,不能使用next()访问。