In my experience, it is not good approach to stack a such large tensor at once.
I recommend you to store (256x256 + 3, 1) size of tensor which indicates a single data.
Storing each data points separately has an advantage of easy managing and fast loading and so forth.
Think about why such a huge datasets (well-known open datasets) manages all data points(images) separately.