首页 > 其他 > 详细

class torch.utils.data.TensorDataset(data_tensor, target_tensor)

时间:2021-06-21 09:54:03      阅读:37      评论:0      收藏:0      [点我收藏+]

class torch.utils.data.TensorDataset(data_tensor, target_tensor)

train_features=torch.tensor([[1.1,2.1]])
train_labels=torch.tensor([[1.1]])
print(torch.utils.data.TensorDataset(train_features,train_labels)[0])   
(tensor([1.1000, 2.1000]), tensor([1.1000]))
该函数将 特征向量和标签打包成一个列表。

class torch.utils.data.TensorDataset(data_tensor, target_tensor)

原文:https://www.cnblogs.com/hahaah/p/14908633.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!