My array X = trainset_array[:, 0] with the shape of (2500, ) and it saved with the 32x32x3 image info.
When I use torch.Tensor(list(ndarray)), X_tensor.size() = torch.Size([2500, 32, 32, 3]), I wonder how can I modify the in_channels=32 to in_channels=3 ?
If I don’t wanna modify the model code, how can I modify the array? Thanks in advance!