weixin_44918105的博客

torch.utils.data.DataLoadertorch.utils.data.Dataset. Datasettorchvision.datasets模块数据集包含许多真实世界视觉数据的对象，如 CIFAR、COCO。在本教程中，我们使用 FashionMNIST 数据集。每个 TorchVision的数据集包含两个参数：transform和 target_transform样本和标签。下载数据集import torchfrom torch import nnfrom torch.ut

在 PyTorch 中，我们使用张量对模型的输入和输出以及模型的参数进行编码。张量初始化import torchimport numpy as np#初始化#直接数据转张量data=[[1,2],[3,4]]x_data=torch.tensor(data)#numpy数组转张量np_array=np.array(data)x_np=torch.from_numpy(np_array)#从另一个张量中来x_ones=torch.ones_like(x_data)#根据给定张量生成与其