So just to rephrase you are saying,that once I get that tensor X,I should simply save it in a npy file .And thanks for the comment,in hindsight I think that makes more sense.
And hence my final dataset will contain about 17,000 npys right?
Also,is it better to downsample the image from 1024 to 256,because I feel that during training having a 1048579 vector might be a problem