首页 > 其他 > 详细

pytorch的torch.utils.data.DataLoader认识

时间:2019-03-27 13:32:24      阅读:137      评论:0      收藏:0      [点我收藏+]

PyTorch中数据读取的一个重要接口是torch.utils.data.DataLoader,该接口定义在dataloader.py脚本中,只要是用PyTorch来训练模型基本都会用到该接口,

该接口主要用来将自定义的数据读取接口的输出或者PyTorch已有的数据读取接口的输入按照batch size封装成Tensor,后续只需要再包装成Variable即可作为模型的输入,

因此该接口有点承上启下的作用,比较重要

pytorch的torch.utils.data.DataLoader认识

原文:https://www.cnblogs.com/demo-deng/p/10606487.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!