batch的理解

mport torch import torch.utils.data as Data BATCH_SZIE = 3 x = torch.linspace(1,10,10) y = torch.linspace(10,1,10) # print(x) # tensor 類型,可以直接轉爲numpy torch_dataset = Data.TensorDataset(x, y) loader
相關文章
相關標籤/搜索