摘要: 张量 张量是一种特殊的数据结构,与数组和矩阵非常相似。 在 PyTorch 中,我们使用张量对模型的输入和输出以及模型的参数进行编码。 张量类似于 NumPy 的 ndarray,此外张量可以在 GPU 或其他硬件加速器上运行。事实上,张量和 NumPy 数组通常可以共享相同的底层内存,无需复制数据 阅读全文
posted @ 2023-04-08 23:59 周XX 阅读(73) 评论(0) 推荐(0) 编辑
摘要: 本文主要介绍机器学习中常见任务的API。 处理数据 PyTorch有两个处理数据的方式:torch.utils.data.DataLoader 和torch.utils.data.Dataset 。 Dataset存储样本及其相应的标签, DataLoader 在Dataset的外层用迭代器进行包装 阅读全文
posted @ 2023-04-08 22:41 周XX 阅读(149) 评论(0) 推荐(0) 编辑