美文网首页
pytorch各种操作

pytorch各种操作

作者: 啊啊啊啊啊1231 | 来源:发表于2020-03-24 14:17 被阅读0次

pytorch为了加快数据提取,定义建立train_data,train_loader

train_data = TensorDataset(torch.tensor(train_data, dtype=torch.int))

train_loader = DataLoader(train_data, batch_size= batch_size,shuffle=True, drop_last=True)

相关文章

网友评论

      本文标题:pytorch各种操作

      本文链接:https://www.haomeiwen.com/subject/jiuoyhtx.html