2023年3月2日
TensorDataset和DataLoader
摘要： 一、TensorDataset 语法：class torch.utils.data.TensorDataset(data_tensor, target_tensor) 作用：包装数据和目标张量（类似Python中的zip（）函数），可通过第一维度索引两个张量恢复数据。故要保证两个tensor的第一维 阅读全文
posted @ 2023-03-02 17:13 Kruskal 阅读(244) 评论(0) 推荐(0) 编辑