测试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)
文章来源:https://www.toymoban.com/news/detail-683322.html
测试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)
文章来源地址https://www.toymoban.com/news/detail-683322.html
到了这里,关于tensordataset 和dataloader取值的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!