tensordataset 和dataloader取值

我的心永远是冰冰哒

已于 2023-08-24 03:09:55 修改

阅读量695

点赞数

文章标签： pytorch 深度学习人工智能

于 2023-08-24 03:05:50 首次发布

本文链接：https://blog.csdn.net/qq_45759229/article/details/132463919

版权

测试1

from torch.utils.data import TensorDataset,DataLoader
import numpy as np 
import torch 

a = np.array([[1,2,3],[2,3,3],[1,1,2],[10,10,10],[100,200,200],[-1,-2,-3]])
print(a)

X = torch.FloatTensor(a)
print(X)

dataset =TensorDataset(X,X)

在这里插入图片描述

测试2

from torch.utils.data import TensorDataset,DataLoader
import numpy as np 
import torch 

a = np.array([[1,2,3],[2,3,3],[1,1,2],[10,10,10],[100,200,200],[-1,-2,-3]])
print(a)

X = torch.FloatTensor(a)
print(X)

dataset =TensorDataset(X,X)

dataloader = DataLoader(dataset,batch_size=4)

在这里插入图片描述