dedell的博客

a = torch.ones(2,2)b = a.numpy()a = numpy.ones(5)b = torch.from_numpy(a)

class DealDataset(Dataset): """ 下载数据、初始化数据，都可以在这里完成 """ def __init__(self): xy = np.loadtxt('../dataSet/diabetes.csv.gz', delimiter=',', dtype=np.float32) # 使用numpy读取数据 ...