qq_43098167的博客

self.add_module()

数据加载Dataset和DataLoader数据集：torch.utils.data.Dataset表示数据集的抽象类，任何自定义的数据集都要继承这个类，并重写相关方法。Pytorch支持两种不同类型的数据集：所有映射类型数据集子类都应该重写__getitem__(self, index) ，支持获取给定键的数据样本。重写__len__(self) ，返回数据集的大小；表示索引到数据样本的映射。迭代类型数据集MarkPyTorch提供的也可以自己定义采样器：自定义时...

torch.cat((a,b), dim=0) 在张量（tensor）的基础维度上拼接在一起torch.stack([a,b], dim=0) 创造一个新的维度

Pytorch 小白散记