TensorDataset()打包后如何拆包
pytorch吧
全部回复
仅看楼主
level 1
yy核动力革民
楼主
用torch.utils.data.TensorDataset()函数把一个数据集打包好了,输出成文件,重新加载文件后发现我不需要用整个数据集,只需要所有数据的特征值,我这里是X_normal,想问下有没有办法扣出来。代码看两个红框的就行
2024年12月09日 12点12分
1
level 1
yy核动力革民
楼主
顶
2024年12月09日 12点12分
2
level 11
sealed_ss
加载到loader里遍历就行
2024年12月09日 13点12分
3
1