Data pipeline for large number of npy files for tensorflow

Question

I currently have a directory containing a large number of numpy arrays with shape (50,50,3) that I wish to use as training, testing, and validation data for my model. Each a