Right, so, for example, your self.labels inside dataset have shape (10000,1)
Then getitem will return single item (1), then you’re applying view(-1, 1), which transforms item to shape (1,1), then batch loader will return batch_num of elements with shape (1, 1)