Scale tensor in dataloader/dataset

Resolved. Just return the int is ok.

label, data = data
data = self.preprocess_data(data)
return data, label