标签: c++ pytorch
我正在尝试使用pyTorch C ++进行一些机器学习。我在.csv文件中有自己的数据集,加载它们后收到例如。具有N个样本和M个特征的[N,M] 2D std :: vector和具有N个标签的[N,] 1D std :: vector。
我的问题是,我应该怎么做才能创建适当的数据集以使用torch :: data :: make_data_loader方法(或其他方法)并学习我的NN。
谢谢!