Thanks. That works.
Another inquiry is about:
return Tensordataset=(torch.FloatTensor(np.array(data)), torch.LongTensor(np.array(labels)))
I convert it in C++
As
Tensordataset[0]=torch::Floattensor(data)
No member named Floattensor
Any suggestion to modify that please ?
Thanks a lot.