About the type of dataset in the function input in libtorch

C++
1

Hi, All

I have a question about the type of dataset defined in function input in libtorch

in pytorch it is
def init(self, labeled, unlabeled):
self.unlabeled = unlabeled

def MnistUnlabel():
raw_dataset = datasets.MNIST(’…/data’, train=True, download=True,
transform=transforms.Compose([
transforms.ToTensor(),
]))
return raw_dataset

May I ask in C++ libtorch how shall I define the type of the dataset unlabel, I tried with torch:data:datasets& unlabel but it is incorrect.

Many thanks.