测试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)
本文转载自: https://blog.csdn.net/qq_45759229/article/details/132463919
版权归原作者 我的心永远是冰冰哒 所有， 如有侵权，请联系我们删除。
版权归原作者 我的心永远是冰冰哒 所有， 如有侵权，请联系我们删除。