how can i covert my data. my target data is (12, 256, 256) in custom datasetclass and (16, 12, 256,256) when calling pytorch data loader. what changes should i do in my customdataclass. i have already did one hot encoding for 12 classes.