Point is that why my dataset class is behaving in two different ways. When loader wants a batch of size 8, it should automatically create a batch of size 8, ain’t it? Why I need to use unsqueeze in the first place?