pytorch
知之之
这个作者很懒,什么都没留下…
展开
-
torch.argsort()
返回能使得数据有序的索引。比如argsort([2,1]),结果是[1,0],即:最小的元素是索引为0的元素,其次是1.原创 2022-01-19 19:38:25 · 3338 阅读 · 1 评论 -
pytorch文档学习——QUICKSTART
0本节代码Working with data 处理数据Creating Models 创建模型本节代码Working with data 处理数据讲述了如何处理数据,两个最关键的部分:torch.utils.data.DataLoader and torch.utils.data.Dataset#前者存储,后者迭代PyTorch 提供领域特定的库,如 TorchText、 TorchVision 和 TorchAudio,所有这些都包含数据集Dataset有俩参数: transform 和原创 2021-06-20 15:10:52 · 241 阅读 · 0 评论