
深度学习
文章平均质量分 90
captainst
这个作者很懒,什么都没留下…
展开
专栏收录文章
- 默认排序
- 最新发布
- 最早发布
- 最多阅读
- 最少阅读
-
有关python的iter,next,yield,和pytorch的dataloader
问题首先,对于一个python数组,我们可以用for,或者next来遍历。其次,而对于pytorch的 torch.utils.data.DataLoader, 也可以使用类似 next(iter(DataLoader)) 的方式遍历地读取dataset的数据。再次,使用yield关键字,也可以起到“遍历”的效果。那么问题来了。这iter,next,yield几个东西之间有什么关联,又有什么区别呢?pytorch的DataLoader又是用的什么样的方式呢?分析先看一个简单例子:fruit =原创 2021-12-08 13:29:14 · 8411 阅读 · 2 评论 -
pytorch的dataset和dataloader
简单说,dataset是数据集,dataloader是加载数据集的工具datasetpytorch提供了多样化的dataset方法。如果你的数据集比较小, xxx和 yyy都可以load到内存里,可以直接使用pytorch的torch.utils.data.TensorDataset:import torchfrom torch.utils import data# build a toy dataset, with a sequence of x and y using y = sin(原创 2021-10-17 15:07:52 · 280 阅读 · 0 评论 -
矩阵操作 - view & permutation
矩阵操作 - view & permutation1. 提出问题2. 解决方案3. 讨论6. 参考1. 提出问题对于一个2D特征图 x∈RM×N×dx \in\mathbb R^{M\times N \times d}x∈RM×N×d,空间尺寸M×NM \times NM×N,将其划分为 sp×sps_p \times s_psp×sp 的窗格,请问用代码如何表示(上面这个问题来自Focal Self-attention for Local-Global Interactions in V原创 2021-10-05 12:32:46 · 211 阅读 · 0 评论